Prevent the point from going through the wall
This commit is contained in:
+2
-2
@@ -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,)
|
||||
|
||||
Reference in New Issue
Block a user