소스 검색

remove import Wall

Rodrigo Perez-Vicente 2 년 전
부모
커밋
7e213bcb3a
1개의 변경된 파일4개의 추가작업 그리고 6개의 파일을 삭제
  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