diff --git a/mujoco_maze/maze_task.py b/mujoco_maze/maze_task.py index b72297e..31c21c9 100644 --- a/mujoco_maze/maze_task.py +++ b/mujoco_maze/maze_task.py @@ -259,7 +259,7 @@ class GoalRewardMultiFall(GoalRewardUMaze): MAZE_SIZE_SCALING: Scaling = Scaling(ant=2.0, point=6.0, swimmer=None) OBSERVE_BLOCKS: bool = True - def __init__(self, scale: float, goal: Tuple[int, int] = (0.0, 3.0)) -> None: + def __init__(self, scale: float, goal: Tuple[int, int] = (0.0, 4.0)) -> None: super().__init__(scale) self.goals = [MazeGoal(np.array([*goal, 0.5]) * scale)] @@ -269,9 +269,9 @@ class GoalRewardMultiFall(GoalRewardUMaze): M = MazeCell.YZ_BLOCK return [ [B, B, B, B, B, B, B, B, B], - [B, E, C, E, E, E, C, E, B], [B, E, C, E, R, E, C, E, B], [B, E, C, E, M, E, C, E, B], + [B, E, C, E, E, E, C, E, B], [B, B, B, C, C, C, B, B, B], [B, B, B, E, E, E, B, B, B], [B, B, B, B, B, B, B, B, B],