eager disconnect

thatguy11325 · Oct 20, 2024 · b8bf0b3 · b8bf0b3
1 parent 73dd67e
commit b8bf0b3
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/pokemonred_puffer/cleanrl_puffer.py b/pokemonred_puffer/cleanrl_puffer.py
@@ -381,9 +381,9 @@ def evaluate(self):
                                     FROM states
                                     """,
                                 ).fetchall()
-                                if all(not reset for reset, env_id in resets if env_id in key_set):
-                                    break
-                                time.sleep(0.5)
+                            if all(not reset for reset, env_id in resets if env_id in key_set):
+                                break
+                            time.sleep(0.5)
                     if self.config.async_wrapper:
                         for key, state in zip(self.event_tracker.keys(), new_states):
                             self.env_recv_queues[key].put(state)