From ddb8a5f3c0b66d6a1121dc1c96cef85f23a6e47a Mon Sep 17 00:00:00 2001 From: kngwyu Date: Fri, 28 May 2021 11:44:11 +0900 Subject: [PATCH] Change the start position of SquareRoom --- mujoco_maze/maze_task.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/mujoco_maze/maze_task.py b/mujoco_maze/maze_task.py index 9e102a9..21b4ce6 100644 --- a/mujoco_maze/maze_task.py +++ b/mujoco_maze/maze_task.py @@ -148,7 +148,7 @@ class DistRewardSimpleRoom(GoalRewardSimpleRoom, DistRewardMixIn): class GoalRewardSquareRoom(GoalRewardUMaze): MAZE_SIZE_SCALING: Scaling = Scaling(4.0, 4.0, 2.0) - def __init__(self, scale: float, goal: Tuple[float, float] = (2.0, 0.0)) -> None: + def __init__(self, scale: float, goal: Tuple[float, float] = (1.0, 0.0)) -> None: super().__init__(scale) self.goals = [MazeGoal(np.array(goal) * scale)] @@ -158,7 +158,7 @@ class GoalRewardSquareRoom(GoalRewardUMaze): return [ [B, B, B, B, B], [B, E, E, E, B], - [B, R, E, E, B], + [B, E, R, E, B], [B, E, E, E, B], [B, B, B, B, B], ]