Tianshou/examples/mujoco/register.py
Minghao Zhang 0b08a41610
move mujoco to examples (#12)
* move mujoco to examples

* fix the import mujoco bug

* flake8

* flake8

* rm __init__.py
2020-04-02 08:49:19 +08:00

28 lines
653 B
Python

from gym.envs.registration import register
def reg():
register(
id='PointMaze-v0',
entry_point='mujoco.point_maze_env:PointMazeEnv',
kwargs={
"maze_size_scaling": 4,
"maze_id": "Maze2",
"maze_height": 0.5,
"manual_collision": True,
"goal": (1, 3),
}
)
register(
id='PointMaze-v1',
entry_point='mujoco.point_maze_env:PointMazeEnv',
kwargs={
"maze_size_scaling": 2,
"maze_id": "Maze2",
"maze_height": 0.5,
"manual_collision": True,
"goal": (1, 3),
}
)