dodox/fancy_gym
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Register environments separately

Browse Source
This commit is contained in:
Mustafa Enes Batur 2023-11-19 22:45:48 +01:00
parent 9aa572271f
commit 7e537447e1
1 changed files with 10 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
fancy_gym/envs/__init__.py
View File

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