miniworld-maze 1.2.0__tar.gz → 1.3.0__tar.gz

{miniworld_maze-1.2.0 → miniworld_maze-1.3.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: miniworld-maze
-Version: 1.2.0
+Version: 1.3.0
 Summary: Multi-room maze environments from the DrStrategy paper. Provides NineRooms-v0, SpiralNineRooms-v0, and TwentyFiveRooms-v0 gymnasium environments.
 Keywords: reinforcement-learning,environment,gymnasium,multi-room-maze,drstrategy,maze-navigation,partial-observability,3d-environments
 Author: Tim Joseph

{miniworld_maze-1.2.0 → miniworld_maze-1.3.0}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "uv_build"
 
 [project]
 name = "miniworld-maze"
-version = "1.2.0"
+version = "1.3.0"
 description = "Multi-room maze environments from the DrStrategy paper. Provides NineRooms-v0, SpiralNineRooms-v0, and TwentyFiveRooms-v0 gymnasium environments."
 readme = "README.md"
 license = {file = "LICENSE"}

{miniworld_maze-1.2.0 → miniworld_maze-1.3.0}/src/miniworld_maze/core/miniworld_gymnasium/base_env.py

@@ -21,6 +21,7 @@ class MiniWorldEnv(UnifiedMiniWorldEnv):
         window_height=600,
         params=DEFAULT_PARAMS,
         domain_rand=False,
+        render_mode=None,
     ):
         """
         Initialize base MiniWorld environment.
@@ -33,6 +34,7 @@ class MiniWorldEnv(UnifiedMiniWorldEnv):
             window_height: Window height for human rendering
             params: Environment parameters for domain randomization
             domain_rand: Whether to enable domain randomization
+            render_mode: Render mode ("human", "rgb_array", or None)
         """
         # Mark this as a base environment (not custom) for background color handling
         self._is_custom_env = False
@@ -49,4 +51,5 @@ class MiniWorldEnv(UnifiedMiniWorldEnv):
             window_height=window_height,
             params=params,
             domain_rand=domain_rand,
+            render_mode=render_mode,
         )

{miniworld_maze-1.2.0 → miniworld_maze-1.3.0}/src/miniworld_maze/core/miniworld_gymnasium/unified_env.py

@@ -63,7 +63,7 @@ class UnifiedMiniWorldEnv(gym.Env):
     both the enhanced features of CustomMiniWorldEnv and the legacy BaseEnv functionality.
     """
 
-    metadata = {"render.modes": ["human", "rgb_array"], "video.frames_per_second": 30}
+    metadata = {"render_modes": ["human", "rgb_array"], "video.frames_per_second": 30}
 
     # Enumeration of possible actions
     class Actions(IntEnum):
@@ -98,6 +98,7 @@ class UnifiedMiniWorldEnv(gym.Env):
         params=DEFAULT_PARAMS,
         domain_rand=False,
         info_obs: Optional[List[ObservationLevel]] = None,
+        render_mode=None,
     ):
         """
         Initialize unified MiniWorld environment.
@@ -114,6 +115,7 @@ class UnifiedMiniWorldEnv(gym.Env):
             params: Environment parameters for domain randomization
             domain_rand: Whether to enable domain randomization
             info_obs: List of observation levels to include in info dictionary
+            render_mode: Render mode ("human", "rgb_array", or None)
         """
         # Store configuration
         self.obs_level = obs_level
@@ -123,6 +125,13 @@ class UnifiedMiniWorldEnv(gym.Env):
         self.params = params
         self.domain_rand = domain_rand
         self.info_obs = info_obs
+        self.render_mode = render_mode
+
+        # Validate render_mode
+        if render_mode is not None and render_mode not in self.metadata["render_modes"]:
+            raise ValueError(
+                f"render_mode must be one of {self.metadata['render_modes']}, got {render_mode}"
+            )
 
         # Setup action space
         self._setup_action_space()

{miniworld_maze-1.2.0 → miniworld_maze-1.3.0}/src/miniworld_maze/environments/base_grid_rooms.py

@@ -46,6 +46,7 @@ class GridRoomsEnvironment(UnifiedMiniWorldEnv):
         agent_mode: Optional[str] = None,
         obs_width: int = DEFAULT_OBS_WIDTH,
         obs_height: int = DEFAULT_OBS_HEIGHT,
+        render_mode=None,
         **kwargs,
     ):
         """
@@ -64,6 +65,7 @@ class GridRoomsEnvironment(UnifiedMiniWorldEnv):
             agent_mode: Agent rendering mode ('triangle', 'circle', 'empty')
             obs_width: Observation width in pixels (defaults to DEFAULT_OBS_WIDTH)
             obs_height: Observation height in pixels (defaults to DEFAULT_OBS_HEIGHT)
+            render_mode: Render mode ("human", "rgb_array", or None)
             **kwargs: Additional arguments passed to parent class
         """
 
@@ -120,6 +122,7 @@ class GridRoomsEnvironment(UnifiedMiniWorldEnv):
             agent_mode=self.agent_mode,
             obs_width=obs_width,
             obs_height=obs_height,
+            render_mode=render_mode,
             **kwargs,
         )