From 5f9110a983ffd39d362dcc899758a43a64da4ec3 Mon Sep 17 00:00:00 2001
From: Simon Alibert <alibert.sim@gmail.com>
Date: Tue, 9 Apr 2024 10:00:16 +0200
Subject: [PATCH] fix

---
 gym_pusht/envs/pusht.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/gym_pusht/envs/pusht.py b/gym_pusht/envs/pusht.py
index a2aa367..607e921 100644
--- a/gym_pusht/envs/pusht.py
+++ b/gym_pusht/envs/pusht.py
@@ -114,7 +114,8 @@ class PushTEnv(gym.Env):
     * TODO:
     """
 
-    metadata = {"render_modes": ["human", "rgb_array"], "render_fps": 10}
+    # metadata = {"render_modes": ["human", "rgb_array"], "render_fps": 10}
+    metadata = {"render_modes": ["rgb_array"], "render_fps": 10}
 
     def __init__(
         self,
@@ -330,7 +331,7 @@ def _render(self, visualize=False):
     def close(self):
         if self.window is not None:
             pygame.display.quit()
-            pygame.quit()
+        pygame.quit()
 
     def teleop_agent(self):
         teleop_agent = collections.namedtuple("TeleopAgent", ["act"])