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

PointMazeMaze -> PointMaze

Browse Source
This commit is contained in:
kngwyu 2020-05-30 15:33:09 +09:00
parent bc40d4cf9b
commit 741746a400
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mujoco_maze/__init__.py
View File

@ -12,7 +12,7 @@ def _get_kwargs(maze_id: str) -> tuple:
for maze_id in MAZE_IDS:
gym.envs.register(
id="AntMaze{}-v0".format(maze_id),
id="Ant{}-v0".format(maze_id),
entry_point="mujoco_maze.ant_maze_env:AntMazeEnv",
kwargs=dict(maze_id=maze_id, maze_size_scaling=8, **_get_kwargs(maze_id)),
max_episode_steps=1000,
@ -21,7 +21,7 @@ for maze_id in MAZE_IDS:
for maze_id in MAZE_IDS:
gym.envs.register(
id="PointMaze{}-v0".format(maze_id),
id="Point{}-v0".format(maze_id),
entry_point="mujoco_maze.point_maze_env:PointMazeEnv",
kwargs=dict(
maze_id=maze_id,

2
mujoco_maze/maze_env.py
View File

@ -603,7 +603,7 @@ def _reward_fn(maze_id: str, dense: str) -> callable:
def _default_goal(maze_id: str) -> np.ndarray:
if maze_id == "Maze" or maze_id == "BlockMaze":
return np.array([0.0, 8.0])
return np.array([0.0, 16.0])
elif maze_id == "Push":
return np.array([0.0, 19.0])
elif maze_id == "Fall":