Fix reset render after call to env.close()

Author: LucasAlegre

mo_gymnasium/envs/deep_sea_treasure/deep_sea_treasure.py

@@ -324,6 +324,8 @@ def close(self):
         if self.window is not None:
             pygame.display.quit()
             pygame.quit()
+            self.window = None
+            self.clock = None
 
 
 if __name__ == "__main__":

mo_gymnasium/envs/four_room/four_room.py

@@ -345,6 +345,8 @@ def close(self):
         if self.window is not None:
             pygame.display.quit()
             pygame.quit()
+            self.window = None
+            self.clock = None
 
 
 def draw_line_dashed(surface, color, start_pos, end_pos, width=1, dash_length=3, exclude_corners=True):

mo_gymnasium/envs/minecart/minecart.py

@@ -692,6 +692,10 @@ def close(self):
         if self.screen is not None:
             pygame.display.quit()
             pygame.quit()
+            self.screen = None
+            self.canvas = None
+            self.clock = None
+            self.last_render_mode_used = None
 
 
 class Mine:

mo_gymnasium/envs/resource_gathering/resource_gathering.py

@@ -305,6 +305,8 @@ def close(self):
         if self.window is not None:
             pygame.display.quit()
             pygame.quit()
+            self.window = None
+            self.clock = None
 
 
 if __name__ == "__main__":