consistent names

thatguy11325 · Jul 8, 2024 · 83ed6f7 · 83ed6f7
1 parent 3a0175a
commit 83ed6f7
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/pokemonred_puffer/cleanrl_puffer.py b/pokemonred_puffer/cleanrl_puffer.py
@@ -257,7 +257,7 @@ def evaluate(self):
                             for count, eid in zip(
                                 self.infos["required_events_count"], self.infos["env_id"]
                             ):
-                                self.events_tracker[eid] = count
+                                self.event_tracker[eid] = count
                             self.infos[k].append(v)
                         else:
                             self.infos[k].append(v)
@@ -286,8 +286,8 @@ def evaluate(self):
                 largest = [
                     x[0]
                     for x in heapq.nlargest(
-                        math.ceil(len(self.events_tracker) * self.config.swarm_keep_pct),
-                        enumerate(self.events_tracker),
+                        math.ceil(len(self.event_tracker) * self.config.swarm_keep_pct),
+                        enumerate(self.event_tracker),
                         key=lambda x: x[1],
                     )
                 ]