use correct key

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()),