Quit Pygame when cliffwalking environment is closed (#1297)

Farama-Foundation · Jan 24, 2025 · 5bc344e · 5bc344e
1 parent 08a28d3
commit 5bc344e
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/gymnasium/envs/toy_text/cliffwalking.py b/gymnasium/envs/toy_text/cliffwalking.py
@@ -345,6 +345,13 @@ def _render_text(self):
         with closing(outfile):
             return outfile.getvalue()
 
+    def close(self):
+        if self.window_surface is not None:
+            import pygame
+
+            pygame.display.quit()
+            pygame.quit()
+
 
 # Elf and stool from https://franuka.itch.io/rpg-snow-tileset
 # All other assets by ____