Merge branch 'main' of https://github.com/Farama-Foundation/MO-Gymnasium

Author: LucasAlegre

mo_gymnasium/envs/deep_sea_treasure/deep_sea_treasure.py

@@ -212,10 +212,16 @@ def _get_map_value(self, pos):
         return self.sea_map[pos[0]][pos[1]]
 
     def _is_valid_state(self, state):
-        if state[0] >= 0 and state[0] <= 10 and state[1] >= 0 and state[1] <= 10:
-            if self._get_map_value(state) != -10:
-                return True
-        return False
+        if self.map_name == "mirrored":
+            if state[0] >= 0 and state[0] <= 10 and state[1] >= 0 and state[1] <= 19:
+                if self._get_map_value(state) != -10:
+                    return True
+            return False
+        else:
+            if state[0] >= 0 and state[0] <= 10 and state[1] >= 0 and state[1] <= 10:
+                if self._get_map_value(state) != -10:
+                    return True
+            return False
 
     def render(self):
         if self.render_mode is None:

mo_gymnasium/envs/four_room/four_room.py

@@ -179,6 +179,9 @@ def step(self, action):
         # can now move
         self.state = (s1, collected)
 
+        if self.render_mode == "human":
+            self.render()
+
         # into a goal cell
         if s1 == self.goal:
             phi = np.ones(len(self.all_shapes), dtype=np.float32)
@@ -206,9 +209,6 @@ def step(self, action):
                 phi = self.features(old_state, action, self.state)
                 return self.state_to_array(self.state), phi, terminated, False, {}
 
-        if self.render_mode == "human":
-            self.render()
-
         # into an empty cell
         return (
             self.state_to_array(self.state),