Add infinite sleep when grading .close() is called

Unity-Technologies · Feb 11, 2019 · 888b370 · 888b370
1 parent 8b1097a
commit 888b370
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/obstacle_tower_env.py b/obstacle_tower_env.py
@@ -217,6 +217,10 @@ def close(self):
         garbage collected or when the program exits.
         """
         self._env.close()
+        if self.is_grading():
+            import time
+            while True:
+                time.sleep(10)
 
     def get_action_meanings(self):
         return self.action_meanings