updated class name for obstructedmaze_v1

Farama-Foundation · Jan 15, 2024 · f694155 · f694155
1 parent 78a9dbc
commit f694155
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 16 deletions.
diff --git a/docs/_scripts/gen_envs_display.py b/docs/_scripts/gen_envs_display.py
@@ -77,19 +77,9 @@ def generate_page(env, limit=-1, base_path=""):
         env_type = key
 
         page = generate_page({"id": key, "list": value})
-        # fp = open(
-        #     os.path.join(
-        #         os.path.dirname(__file__), "..", "environments", env_type, "list.html"
-        #     ),
-        #     "w",
-        #     encoding="utf-8",
-        # )
         fp = open(
             os.path.join(
-                os.path.dirname(__file__),
-                "..",
-                "environments/minigrid",
-                f"{env_type}.md",
+                os.path.dirname(__file__), "..", "environments", env_type, "list.html"
             ),
             "w",
             encoding="utf-8",

diff --git a/minigrid/__init__.py b/minigrid/__init__.py
@@ -478,7 +478,7 @@ def register_minigrid_envs():
 
     register(
         id="MiniGrid-ObstructedMaze-2Dlhb-v1",
-        entry_point="minigrid.envs.obstructedmaze_v1:ObstructedMaze_Full",
+        entry_point="minigrid.envs:ObstructedMaze_Full_V1",
         kwargs={
             "agent_room": (2, 1),
             "key_in_box": True,
@@ -490,7 +490,7 @@ def register_minigrid_envs():
 
     register(
         id="MiniGrid-ObstructedMaze-1Q-v1",
-        entry_point="minigrid.envs.obstructedmaze_v1:ObstructedMaze_Full",
+        entry_point="minigrid.envs:ObstructedMaze_Full_V1",
         kwargs={
             "agent_room": (1, 1),
             "key_in_box": True,
@@ -502,7 +502,7 @@ def register_minigrid_envs():
 
     register(
         id="MiniGrid-ObstructedMaze-2Q-v1",
-        entry_point="minigrid.envs.obstructedmaze_v1:ObstructedMaze_Full",
+        entry_point="minigrid.envs:ObstructedMaze_Full_V1",
         kwargs={
             "agent_room": (2, 1),
             "key_in_box": True,
@@ -514,7 +514,7 @@ def register_minigrid_envs():
 
     register(
         id="MiniGrid-ObstructedMaze-Full-v1",
-        entry_point="minigrid.envs.obstructedmaze_v1:ObstructedMaze_Full",
+        entry_point="minigrid.envs:ObstructedMaze_Full_V1",
     )
 
     # Playground

diff --git a/minigrid/envs/__init__.py b/minigrid/envs/__init__.py
@@ -20,6 +20,7 @@
     ObstructedMaze_Full,
     ObstructedMazeEnv,
 )
+from minigrid.envs.obstructedmaze_v1 import ObstructedMaze_Full_V1
 from minigrid.envs.playground import PlaygroundEnv
 from minigrid.envs.putnear import PutNearEnv
 from minigrid.envs.redbluedoors import RedBlueDoorEnv

diff --git a/minigrid/envs/obstructedmaze_v1.py b/minigrid/envs/obstructedmaze_v1.py
@@ -6,7 +6,7 @@
 from minigrid.envs.obstructedmaze import ObstructedMazeEnv
 
 
-class ObstructedMaze_Full(ObstructedMazeEnv):
+class ObstructedMaze_Full_V1(ObstructedMazeEnv):
     """
     A blue ball is hidden in one of the 4 corners of a 3x3 maze. Doors
     are locked, doors are obstructed by a ball and keys are hidden in