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

Fix max episode steps

Browse Source
This commit is contained in:
Mustafa Enes Batur 2023-11-20 12:38:27 +01:00
parent 079225cd1b
commit 62fecda9d6
1 changed files with 12 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
fancy_gym/envs/__init__.py
View File

@ -291,13 +291,23 @@ register(
)
# Air Hockey environments
for env_mode in ["tournament", "7dof-hit", "7dof-defend", "3dof-hit", "3dof-defend"]:
for env_mode in ["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=500,
add_mp_types=[],
kwargs={
'env_mode': env_mode
}
)
register(
id=f'fancy/AirHockey-tournament-v0',
entry_point='fancy_gym.envs.mujoco:AirHockeyEnv',
max_episode_steps=15000,
add_mp_types=[],
kwargs={
'env_mode': 'tournament'
}
)