fix

huggingface · Apr 9, 2024 · 5f9110a · 5f9110a
1 parent 6c98935
commit 5f9110a
Showing 1 changed file with 3 additions and 2 deletions.
diff --git 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"])