fancy_gym/alr_envs/__init__.py

17 lines
326 B
Python
Raw Normal View History

2020-08-28 18:31:06 +02:00
from gym.envs.registration import register
register(
id='ALRReacher-v0',
entry_point='alr_envs.mujoco:ALRReacherEnv',
max_episode_steps=1000,
)
register(
id='SimpleReacher-v0',
entry_point='alr_envs.classic_control:SimpleReacherEnv',
max_episode_steps=200,
kwargs={
2020-08-31 15:51:47 +02:00
"n_links": 3,
2020-08-28 18:31:06 +02:00
}
)