Add new mujoco envs to __init__

MushroomRL · Jan 2, 2025 · 4a57acc · 4a57acc
1 parent 25e2ae3
commit 4a57acc
Show file tree

Hide file tree

Showing 5 changed files with 12 additions and 4 deletions.
diff --git a/examples/mujoco_ant_ppo.py b/examples/mujoco_ant_ppo.py
@@ -6,7 +6,7 @@
 
 from mushroom_rl.algorithms.actor_critic import PPO
 from mushroom_rl.core import Core, Logger
-from mushroom_rl.environments.mujoco_envs.ant import Ant
+from mushroom_rl.environments import Ant
 from mushroom_rl.policy import GaussianTorchPolicy
 
 from tqdm import trange

diff --git a/examples/mujoco_half_cheetah_ppo.py b/examples/mujoco_half_cheetah_ppo.py
@@ -6,7 +6,7 @@
 
 from mushroom_rl.algorithms.actor_critic import PPO
 from mushroom_rl.core import Core, Logger
-from mushroom_rl.environments.mujoco_envs.half_cheetah import HalfCheetah
+from mushroom_rl.environments import HalfCheetah
 from mushroom_rl.policy import GaussianTorchPolicy
 
 from tqdm import trange

diff --git a/examples/mujoco_hopper_ppo.py b/examples/mujoco_hopper_ppo.py
@@ -6,7 +6,7 @@
 
 from mushroom_rl.algorithms.actor_critic import PPO
 from mushroom_rl.core import Core, Logger
-from mushroom_rl.environments.mujoco_envs.hopper import Hopper
+from mushroom_rl.environments import Hopper
 from mushroom_rl.policy import GaussianTorchPolicy
 
 from tqdm import trange

diff --git a/examples/mujoco_walker_2d_ppo.py b/examples/mujoco_walker_2d_ppo.py
@@ -6,7 +6,7 @@
 
 from mushroom_rl.algorithms.actor_critic import PPO
 from mushroom_rl.core import Core, Logger
-from mushroom_rl.environments.mujoco_envs.walker_2d import Walker2D
+from mushroom_rl.environments import Walker2D
 from mushroom_rl.policy import GaussianTorchPolicy
 
 from tqdm import trange

diff --git a/mushroom_rl/environments/mujoco_envs/__init__.py b/mushroom_rl/environments/mujoco_envs/__init__.py
@@ -1,8 +1,16 @@
 from .ball_in_a_cup import BallInACup
 from .air_hockey import AirHockeyHit, AirHockeyDefend, AirHockeyPrepare, AirHockeyRepel
+from .ant import Ant
+from .half_cheetah import HalfCheetah
+from .hopper import Hopper
+from .walker_2d import Walker2D
 
 BallInACup.register()
 AirHockeyHit.register()
 AirHockeyDefend.register()
 AirHockeyPrepare.register()
 AirHockeyRepel.register()
+Ant.register()
+HalfCheetah.register()
+Hopper.register()
+Walker2D.register()