revert close change

huggingface · Apr 9, 2024 · 7150dea · 7150dea
1 parent 35e4bc0
commit 7150dea
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/gym_pusht/envs/pusht.py b/gym_pusht/envs/pusht.py
@@ -335,7 +335,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"])