Prevent the point from going through the wall

This commit is contained in:
kngwyu
2020-05-30 19:48:02 +09:00
parent 741746a400
commit 348e975e60
8 changed files with 113 additions and 94 deletions
+2 -2
View File
@@ -5,11 +5,11 @@ import pytest
@pytest.mark.parametrize("maze_id", mujoco_maze.MAZE_IDS)
def test_ant_maze(maze_id):
env = gym.make("AntMaze{}-v0".format(maze_id))
env = gym.make("Ant{}-v0".format(maze_id))
assert env.reset().shape == (30,)
@pytest.mark.parametrize("maze_id", mujoco_maze.MAZE_IDS)
def test_point_maze(maze_id):
env = gym.make("PointMaze{}-v0".format(maze_id))
env = gym.make("Point{}-v0".format(maze_id))
assert env.reset().shape == (7,)