From 85fbb1c7f258d2204e4493686a6db4fd2503dbbd Mon Sep 17 00:00:00 2001
From: thatguy11325 <148832074+thatguy11325@users.noreply.github.com>
Date: Wed, 10 Jul 2024 16:09:45 -0400
Subject: [PATCH] use correct key

---
 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 c055a03..1ddbc71 100644
--- a/pokemonred_puffer/cleanrl_puffer.py
+++ b/pokemonred_puffer/cleanrl_puffer.py
@@ -285,7 +285,7 @@ def evaluate(self):
             ):
                 # collect the top swarm_keep_pct % of the envs in the batch
                 largest = [
-                    x[1][1]
+                    x[1][0]
                     for x in heapq.nlargest(
                         math.ceil(len(self.event_tracker) * self.config.swarm_keep_pct),
                         enumerate(self.event_tracker.items()),