Browse Source

remove import Wall

Rodrigo Perez-Vicente 2 years ago
parent
commit
7e213bcb3a
1 changed files with 4 additions and 6 deletions
  1. 4 6
      gym_minigrid/__init__.py

+ 4 - 6
gym_minigrid/__init__.py

@@ -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