(env_name: str)
| 11 | |
| 12 | |
| 13 | def get_env(env_name: str): |
| 14 | if env_name == "pushT": |
| 15 | return PushT() |
| 16 | elif env_name == "hopper": |
| 17 | return Hopper() |
| 18 | elif env_name == "humanoidstandup": |
| 19 | return HumanoidStandup() |
| 20 | elif env_name == "humanoidrun": |
| 21 | return HumanoidRun() |
| 22 | elif env_name == "humanoidtrack": |
| 23 | return HumanoidTrack() |
| 24 | elif env_name == "walker2d": |
| 25 | return Walker2d() |
| 26 | elif env_name == "cartpole": |
| 27 | return Cartpole() |
| 28 | elif env_name == "car2d": |
| 29 | return Car2d() |
| 30 | elif env_name in ["ant", "halfcheetah"]: |
| 31 | return brax_envs.get_environment(env_name=env_name, backend="positional") |
| 32 | else: |
| 33 | raise ValueError(f"Unknown environment: {env_name}") |
nothing calls this directly
no test coverage detected