Implement close

This commit is contained in:
kngwyu
2020-09-17 16:55:52 +09:00
parent 8f630b58c2
commit fe06ba7c5b
3 changed files with 11 additions and 1 deletions
+1 -1
View File
@@ -24,7 +24,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, -4.0)]})
assert env.reset().shape == (7,)
s, _, _, _ = env.step(env.action_space.sample())
assert s.shape == (7,)