Skip to content

Fish

mujorax.FishSwimEnv

Bases: MjxPlaygroundEnv

DM Control FishSwim.

Free-swimming fish in a 3D water tank; dense reward for swimming toward a randomised target.

Parameters:

Name Type Description Default
config MjxPlaygroundConfig

Static configuration. Defaults to MjxPlaygroundConfig().

required
Source code in mujorax/envs/fish.py
Python
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
class FishSwimEnv(MjxPlaygroundEnv):
    """
    DM Control `FishSwim`.

    Free-swimming fish in a 3D water tank; dense reward for swimming
    toward a randomised target.

    Parameters
    ----------
    config : MjxPlaygroundConfig (optional)
        Static configuration. Defaults to `MjxPlaygroundConfig()`.
    """

    _PLAYGROUND_NAME = "FishSwim"