miniworld-maze 1.3.0__py3-none-any.whl → 1.3.2__py3-none-any.whl

miniworld_maze/core/miniworld_gymnasium/unified_env.py

@@ -349,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
@@ -584,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:
@@ -604,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:
@@ -613,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

@@ -230,10 +230,14 @@ class GridRoomsEnvironment(UnifiedMiniWorldEnv):
         obs, reward, terminated, truncated, info = super().step(action)
 
         # Check if goal is achieved
-        if self._is_goal_achieved():
+        goal_achieved = self._is_goal_achieved()
+        if goal_achieved:
             terminated = True
             reward = 1.0  # Positive reward for achieving goal
 
+        # Add success indicator to info dictionary
+        info["success"] = 1.0 if goal_achieved else 0.0
+
         # Add agent and goal positions to info dictionary
         agent_pos = self.agent.pos
         info["agent_position"] = np.array([agent_pos[0], agent_pos[2]])  # x, z

{miniworld_maze-1.3.0.dist-info → miniworld_maze-1.3.2.dist-info}/METADATA

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

@@ -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=7A4sAmMpopLfKMXtZ4jzbMlFJbzeu9Abv7hGYVbvOIA,46925
+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=--tm6t9cHn444xJX9N3Md-4YApK38a2dQi7mnggevbI,14885
+miniworld_maze/environments/base_grid_rooms.py,sha256=ZYB2uTEXhifbXdg2qqJ9LC6r57_kjDfgvYqW7qoWmHg,15031
 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.3.0.dist-info/WHEEL,sha256=NHRAbdxxzyL9K3IO2LjmlNqKSyPZnKv2BD16YYVKo18,79
-miniworld_maze-1.3.0.dist-info/METADATA,sha256=qKeYDRC9Evc7PtegEs0PVFYy7fMgKlqNZPiFf3GnZkk,9657
-miniworld_maze-1.3.0.dist-info/RECORD,,
+miniworld_maze-1.3.2.dist-info/WHEEL,sha256=Jb20R3Ili4n9P1fcwuLup21eQ5r9WXhs4_qy7VTrgPI,79
+miniworld_maze-1.3.2.dist-info/METADATA,sha256=PDoE_djeG02mqcAxRAxLSPlqDZbvW5xynx9O1QAA0g4,9657
+miniworld_maze-1.3.2.dist-info/RECORD,,

{miniworld_maze-1.3.0.dist-info → miniworld_maze-1.3.2.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: uv 0.8.14
+Generator: uv 0.8.15
 Root-Is-Purelib: true
 Tag: py3-none-any