dodox/mujoco_maze
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Closed troom

Browse Source
This commit is contained in:
kngwyu 2020-09-21 13:28:34 +09:00
parent 698d0acd94
commit e2932b5e4b
2 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
mujoco_maze/maze_task.py
View File

@ -249,7 +249,7 @@ class GoalRewardTRoom(MazeTask):
def __init__( def __init__(
self, self,
scale: float, scale: float,
goals: List[Tuple[float, float]] = [(2.0, -4.0)], goals: List[Tuple[float, float]] = [(2.0, -3.0)],
) -> None: ) -> None:
super().__init__(scale) super().__init__(scale)
self.goals = [] self.goals = []
@ -267,7 +267,6 @@ class GoalRewardTRoom(MazeTask):
E, B, R = MazeCell.EMPTY, MazeCell.BLOCK, MazeCell.ROBOT E, B, R = MazeCell.EMPTY, MazeCell.BLOCK, MazeCell.ROBOT
return [ return [
[B, B, B, B, B, B, B], [B, B, B, B, B, B, B],
[B, E, E, E, E, E, B],
[B, E, E, B, E, E, B], [B, E, E, B, E, E, B],
[B, E, E, B, E, E, B], [B, E, E, B, E, E, B],
[B, E, B, B, B, E, B], [B, E, B, B, B, E, B],

2
tests/test_envs.py
View File

@ -28,7 +28,7 @@ def test_point_maze(maze_id):
@pytest.mark.parametrize("v", [0, 1]) @pytest.mark.parametrize("v", [0, 1])
def test_maze_args(v): 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,) assert env.reset().shape == (7,)
s, _, _, _ = env.step(env.action_space.sample()) s, _, _, _ = env.step(env.action_space.sample())
assert s.shape == (7,) assert s.shape == (7,)