From 18a1569a45b2b7ed5db836cdde3c1f87d096becc Mon Sep 17 00:00:00 2001 From: Yuji Kanagawa Date: Wed, 22 Sep 2021 11:30:16 +0900 Subject: [PATCH] Fixe NoRewardSquareRoom --- 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 21b4ce6..e307d74 100644 --- a/mujoco_maze/maze_task.py +++ b/mujoco_maze/maze_task.py @@ -146,7 +146,7 @@ class DistRewardSimpleRoom(GoalRewardSimpleRoom, DistRewardMixIn): class GoalRewardSquareRoom(GoalRewardUMaze): - MAZE_SIZE_SCALING: Scaling = Scaling(4.0, 4.0, 2.0) + MAZE_SIZE_SCALING: Scaling = Scaling(2.0, 4.0, 2.0) def __init__(self, scale: float, goal: Tuple[float, float] = (1.0, 0.0)) -> None: super().__init__(scale) @@ -164,7 +164,7 @@ class GoalRewardSquareRoom(GoalRewardUMaze): ] -class NoRewardSquareRoom(GoalRewardSimpleRoom): +class NoRewardSquareRoom(GoalRewardSquareRoom): def __init__(self, scale: float) -> None: super().__init__(scale)