diff --git a/mujoco_maze/__init__.py b/mujoco_maze/__init__.py index 67e0e0a..a50c1f5 100644 --- a/mujoco_maze/__init__.py +++ b/mujoco_maze/__init__.py @@ -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, diff --git a/mujoco_maze/maze_env.py b/mujoco_maze/maze_env.py index 719b5c5..fed6c67 100644 --- a/mujoco_maze/maze_env.py +++ b/mujoco_maze/maze_env.py @@ -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":