From 1f76669aa374f1f46e380f81929a2c2b321cb772 Mon Sep 17 00:00:00 2001
From: thatguy11325 <148832074+thatguy11325@users.noreply.github.com>
Date: Sat, 26 Oct 2024 23:07:45 -0400
Subject: [PATCH] Was a tad too eager on the delete

---
 pokemonred_puffer/cleanrl_puffer.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/pokemonred_puffer/cleanrl_puffer.py b/pokemonred_puffer/cleanrl_puffer.py
index 4347e60..f825f56 100644
--- a/pokemonred_puffer/cleanrl_puffer.py
+++ b/pokemonred_puffer/cleanrl_puffer.py
@@ -335,12 +335,12 @@ def evaluate(self):
                             self.early_stop = True
                             break
                     for event in to_delete:
-                        del self.config.early_stop[event]
                         print(
                             f"Satisified early stopping constraint for {event} within "
                             f"{self.config.early_stop[event]} minutes. "
                             f"Event found n {self.profile.uptime // 60} minutes"
                         )
+                        del self.config.early_stop[event]
 
                 # V2 implementation
                 # check if we have a new highest required_count with N save states available