Register environments separately

2023-11-19 22:45:48 +01:00 · 2023-11-19 22:45:48 +01:00 · 7e537447e1
commit 7e537447e1
parent 9aa572271f
1 changed files with 10 additions and 5 deletions
--- a/fancy_gym/envs/init.py
+++ b/fancy_gym/envs/init.py
@ -290,8 +290,13 @@ register(
    }
 )

-register(
-    id='fancy/AirHockey-v0',
+# Air Hockey environments
+for env_mode in ["tournament", "7dof-hit", "7dof-defend", "3dof-hit", "3dof-defend"]:
+    register(
+        id=f'fancy/AirHockey-{env_mode}-v0',
        entry_point='fancy_gym.envs.mujoco:AirHockeyEnv',
-    max_episode_steps=45000
-)
+        max_episode_steps=45000,
+        kwargs={
+            'env_mode': env_mode
+        }
+    )