From 3ed517790635e545b17efeeed38fa2bcf7b8d86a Mon Sep 17 00:00:00 2001 From: kngwyu Date: Thu, 24 Sep 2020 17:41:49 +0900 Subject: [PATCH] Revert 2444d09 --- mujoco_maze/maze_task.py | 3 +-- tests/test_envs.py | 2 +- 2 files changed, 2 insertions(+), 3 deletions(-) diff --git a/mujoco_maze/maze_task.py b/mujoco_maze/maze_task.py index 52afbbe..9d36a1e 100644 --- a/mujoco_maze/maze_task.py +++ b/mujoco_maze/maze_task.py @@ -249,7 +249,7 @@ class GoalRewardTRoom(MazeTask): def __init__( self, scale: float, - goals: List[Tuple[float, float]] = [(2.0, -4.0)], + goals: List[Tuple[float, float]] = [(2.0, -3.0)], ) -> None: super().__init__(scale) self.goals = [] @@ -270,7 +270,6 @@ class GoalRewardTRoom(MazeTask): [B, E, E, B, E, E, B], [B, E, E, B, E, E, B], [B, E, B, B, B, E, B], - [B, E, E, E, E, E, B], [B, E, E, R, E, E, B], [B, B, B, B, B, B, B], ] diff --git a/tests/test_envs.py b/tests/test_envs.py index 76e8190..6e925a1 100644 --- a/tests/test_envs.py +++ b/tests/test_envs.py @@ -28,7 +28,7 @@ def test_point_maze(maze_id): @pytest.mark.parametrize("v", [0, 1]) def test_maze_args(v): - env = gym.make(f"PointTRoom-v{v}", task_kwargs={"goals": [(-2.0, -4.0)]}) + env = gym.make(f"PointTRoom-v{v}", task_kwargs={"goals": [(-2.0, -3.0)]}) assert env.reset().shape == (7,) s, _, _, _ = env.step(env.action_space.sample()) assert s.shape == (7,)