|
@@ -3,8 +3,6 @@ from gym_minigrid import wrappers
|
|
|
from gym_minigrid import minigrid
|
|
|
from gym_minigrid import roomgrid
|
|
|
|
|
|
-from gym_minigrid.minigrid import Wall
|
|
|
-
|
|
|
|
|
|
def register_minigrid_envs():
|
|
|
# BlockedUnlockPickup
|
|
@@ -47,25 +45,25 @@ def register_minigrid_envs():
|
|
|
register(
|
|
|
id="MiniGrid-SimpleCrossingS9N1-v0",
|
|
|
entry_point="gym_minigrid.envs:CrossingEnv",
|
|
|
- kwargs={"size": 9, "num_crossings": 1, "obstacle_type": Wall},
|
|
|
+ kwargs={"size": 9, "num_crossings": 1, "obstacle_type": minigrid.Wall},
|
|
|
)
|
|
|
|
|
|
register(
|
|
|
id="MiniGrid-SimpleCrossingS9N2-v0",
|
|
|
entry_point="gym_minigrid.envs:CrossingEnv",
|
|
|
- kwargs={"size": 9, "num_crossings": 2, "obstacle_type": Wall},
|
|
|
+ kwargs={"size": 9, "num_crossings": 2, "obstacle_type": minigrid.Wall},
|
|
|
)
|
|
|
|
|
|
register(
|
|
|
id="MiniGrid-SimpleCrossingS9N3-v0",
|
|
|
entry_point="gym_minigrid.envs:CrossingEnv",
|
|
|
- kwargs={"size": 9, "num_crossings": 3, "obstacle_type": Wall},
|
|
|
+ kwargs={"size": 9, "num_crossings": 3, "obstacle_type": minigrid.Wall},
|
|
|
)
|
|
|
|
|
|
register(
|
|
|
id="MiniGrid-SimpleCrossingS11N5-v0",
|
|
|
entry_point="gym_minigrid.envs:CrossingEnv",
|
|
|
- kwargs={"size": 11, "num_crossings": 5, "obstacle_type": Wall},
|
|
|
+ kwargs={"size": 11, "num_crossings": 5, "obstacle_type": minigrid.Wall},
|
|
|
)
|
|
|
|
|
|
# DistShift
|