瀏覽代碼

Fixed issue when generating documentation (#414)

Bolun Dai 1 年之前
父節點
當前提交
1eb3617036

+ 1 - 1
.github/workflows/build-docs-dev.yml

@@ -21,7 +21,7 @@ jobs:
         run: pip install -r docs/requirements.txt
 
       - name: Register Envs
-        run: pip install -e .
+        run: pip install -e .[wfc]
 
       - name: Build Envs Docs
         run: python docs/_scripts/gen_env_docs.py

+ 1 - 1
.github/workflows/build-docs-version.yml

@@ -26,7 +26,7 @@ jobs:
         run: pip install -r docs/requirements.txt
 
       - name: Register Envs
-        run: pip install -e .
+        run: pip install -e .[wfc]
 
       - name: Build Envs Docs
         run: python docs/_scripts/gen_env_docs.py

+ 1 - 1
.github/workflows/manual-build-docs-version.yml

@@ -37,7 +37,7 @@ jobs:
         run: pip install -r docs/requirements.txt
 
       - name: Register Envs
-        run: pip install -e .
+        run: pip install -e .[wfc]
 
       - name: Build Envs Docs
         run: python docs/_scripts/gen_env_docs.py

+ 4 - 4
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

+ 1 - 0
minigrid/envs/__init__.py

@@ -20,6 +20,7 @@ from minigrid.envs.obstructedmaze import (
     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

+ 1 - 1
minigrid/envs/obstructedmaze_v1.py

@@ -6,7 +6,7 @@ from minigrid.core.world_object import Ball, Box, Key
 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

+ 5 - 0
minigrid/minigrid_env.py

@@ -200,6 +200,11 @@ class MiniGridEnv(gym.Env):
 
         output = ""
 
+        # check if self.agent_pos & self.agent_dir is None
+        # should not be after env is reset
+        if self.agent_pos is None:
+            return super().__str__()
+
         for j in range(self.grid.height):
             for i in range(self.grid.width):
                 if i == self.agent_pos[0] and j == self.agent_pos[1]: