miniworld-maze 1.2.0__py3-none-any.whl → 1.3.1__py3-none-any.whl

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/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()
@@ -340,8 +349,8 @@ class UnifiedMiniWorldEnv(gym.Env):
             for obs_level in self.info_obs:
                 # Generate observation with the specified level
                 info_obs = self._generate_observation(observation_level=obs_level)
-                # Use the observation level name as key
-                info[str(obs_level)] = info_obs
+                # Use the observation level enum as key
+                info[obs_level] = info_obs
 
         # Return first observation with info dict for Gymnasium compatibility
         return obs, info
@@ -575,19 +584,14 @@ class UnifiedMiniWorldEnv(gym.Env):
 
     def _calculate_step_results(self, observation):
         """Calculate reward, termination, and info for step."""
-        # Generate topdown view for info if needed
-        topdown = None
-        if self.obs_level != 2:  # Not TOP_DOWN_FULL
-            topdown = self.render_top_view(POMDP=False, frame_buffer=self.topdown_fb)
-
         # Generate additional observations for info dictionary if specified
         info = {}
         if self.info_obs is not None:
             for obs_level in self.info_obs:
                 # Generate observation with the specified level
                 info_obs = self._generate_observation(observation_level=obs_level)
-                # Use the observation level name as key
-                info[str(obs_level)] = info_obs
+                # Use the observation level enum as key
+                info[obs_level] = info_obs
 
         # Check termination
         if self.step_count >= self.max_episode_steps:
@@ -595,8 +599,7 @@ class UnifiedMiniWorldEnv(gym.Env):
             reward = 0
             info.update(
                 {
-                    "pos": self.agent.pos,
-                    "mdp_view": topdown if topdown is not None else observation,
+                    "pos": np.array([self.agent.pos[0], self.agent.pos[2]]),
                 }
             )
         else:
@@ -604,8 +607,7 @@ class UnifiedMiniWorldEnv(gym.Env):
             terminated = False
             info.update(
                 {
-                    "pos": self.agent.pos,
-                    "mdp_view": topdown if topdown is not None else observation,
+                    "pos": np.array([self.agent.pos[0], self.agent.pos[2]]),
                 }
             )
 

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,
         )
 

{miniworld_maze-1.2.0.dist-info → miniworld_maze-1.3.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: miniworld-maze
-Version: 1.2.0
+Version: 1.3.1
 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.dist-info → miniworld_maze-1.3.1.dist-info}/RECORD

@@ -3,7 +3,7 @@ miniworld_maze/core/__init__.py,sha256=5BA4WKXQjrG55TNaEid2JGrnf1KQniJZ1HhRqovM1
 miniworld_maze/core/constants.py,sha256=UXqmuvsT9ww1GljW74lWPPP9XxVmvti-8JfM_RBSoWQ,4314
 miniworld_maze/core/miniworld_gymnasium/README.md,sha256=kkZgkRKBdgixpot3uHuiBFlRIKRFIVBVfXwu68XTEv0,74
 miniworld_maze/core/miniworld_gymnasium/__init__.py,sha256=ALOqr4SCVk9I9bmxyoU0Du8LUPE_57jIyZuV4QhOcBc,159
-miniworld_maze/core/miniworld_gymnasium/base_env.py,sha256=heoJxEw_GGEHSzU75MWgqRkWARY3lgTgwNE9Zp1P_YA,1795
+miniworld_maze/core/miniworld_gymnasium/base_env.py,sha256=K2L5r4gG2k4fEknFqRJ7f-b2S0Uj5QhanS-MLi6ncVc,1927
 miniworld_maze/core/miniworld_gymnasium/entities/__init__.py,sha256=gbMJcFeasvlRSIuXWyOYO-MhW0JKKY5ZNxF5TMsWJ5c,262
 miniworld_maze/core/miniworld_gymnasium/entities/agent.py,sha256=U3Zt2Hk21rJSdzPCBj46-N930n7XAkA9qo_fKbezn2k,3153
 miniworld_maze/core/miniworld_gymnasium/entities/base_entity.py,sha256=ox2k3rf64Y2mp4S0mcpNpPUg0QeUuJ7LVoduvBuLXDg,3657
@@ -260,18 +260,18 @@ miniworld_maze/core/miniworld_gymnasium/textures/white_1.png,sha256=wRrgs92I_Ids
 miniworld_maze/core/miniworld_gymnasium/textures/wood_1.png,sha256=XRZyIN34HFo14olbxRcsHGrzCAFqUlowc6nLR22IFBE,184713
 miniworld_maze/core/miniworld_gymnasium/textures/wood_2.png,sha256=qSDHB-ZO11JJLQuiQse-0edpbuTg1YO-eIBhdTvNUhc,93121
 miniworld_maze/core/miniworld_gymnasium/textures/wood_planks_1.png,sha256=E4SNN1s4yOtkLfZFQy905eip6KvDWnnPUrpS82FxMAg,847259
-miniworld_maze/core/miniworld_gymnasium/unified_env.py,sha256=B7kfG6axiL05o3kLiBvE6O1Q7kB8PaLdlAt1-Sksv0I,46529
+miniworld_maze/core/miniworld_gymnasium/unified_env.py,sha256=_xkznGIzXQpZOULgK-XFZaXZbhsouEupByf-4cfZLZc,46607
 miniworld_maze/core/miniworld_gymnasium/utils.py,sha256=9cfpg4qYz-Esxvu8nTMPFJc-Tl0TRxTrX6cfg0YuK_o,1007
 miniworld_maze/core/miniworld_gymnasium/wrappers.py,sha256=cD0nGSJYNU96zoWv63aEiKd986POhtHfGGEpNpRL5ec,122
 miniworld_maze/core/observation_types.py,sha256=Co8mEIXzIgk0MLx6tqeBd1EE0PuZOL1gbZwobiEde08,1316
 miniworld_maze/environments/__init__.py,sha256=DKld5MQU7x9eNL6BlxIettA44bCiIn2zIpYECDCNxoQ,331
-miniworld_maze/environments/base_grid_rooms.py,sha256=j4LnyUv0YjHlCLfjqX3fyoXUOOyCr4e3_Pdj-sM3FHo,14753
+miniworld_maze/environments/base_grid_rooms.py,sha256=--tm6t9cHn444xJX9N3Md-4YApK38a2dQi7mnggevbI,14885
 miniworld_maze/environments/factory.py,sha256=Zk26JawsMgSLMgvSnQxFhQCD8yMH76HgqFtogzWFfqY,1333
 miniworld_maze/environments/nine_rooms.py,sha256=Ct96cKtSt1_nLNI5RBUhwqdNUQq1rHfBJ3aB5Igbdow,1794
 miniworld_maze/environments/spiral_nine_rooms.py,sha256=a_pUuv-ghez8h76Z7YsHkQoLXsQ-w9azKLjEjO4uKmA,1749
 miniworld_maze/environments/twenty_five_rooms.py,sha256=MewKPDHDilscQGTT3aGRrSHvo4uFgHHAOrnJMrHaezQ,2598
 miniworld_maze/tools/__init__.py,sha256=XiReXrJIcBKvDVyPZrKPq1mckJs0_WC7q_RmdXXcKHs,55
 miniworld_maze/utils.py,sha256=HTOkfRq72oOC844gVXjWMH1ox7wdSxfCS6oTrWBw05Q,7523
-miniworld_maze-1.2.0.dist-info/WHEEL,sha256=NHRAbdxxzyL9K3IO2LjmlNqKSyPZnKv2BD16YYVKo18,79
-miniworld_maze-1.2.0.dist-info/METADATA,sha256=QHJBCKSCHQr2u-Utcck1qMOffK1Cc6T0sV2k6_H7OAE,9657
-miniworld_maze-1.2.0.dist-info/RECORD,,
+miniworld_maze-1.3.1.dist-info/WHEEL,sha256=NHRAbdxxzyL9K3IO2LjmlNqKSyPZnKv2BD16YYVKo18,79
+miniworld_maze-1.3.1.dist-info/METADATA,sha256=P-TEotub0dhvLm08sS1HWkRQWqF3HDzXOlUpVEPyPQs,9657
+miniworld_maze-1.3.1.dist-info/RECORD,,