From 2b3ac4d13b79be0c844195fc66c5c4ca862f256f Mon Sep 17 00:00:00 2001
From: thatguy11325 <148832074+thatguy11325@users.noreply.github.com>
Date: Tue, 9 Jul 2024 18:16:08 -0400
Subject: [PATCH] Fix printing again

---
 pokemonred_puffer/cleanrl_puffer.py | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/pokemonred_puffer/cleanrl_puffer.py b/pokemonred_puffer/cleanrl_puffer.py
index b743ce3..9c8f8d1 100644
--- a/pokemonred_puffer/cleanrl_puffer.py
+++ b/pokemonred_puffer/cleanrl_puffer.py
@@ -305,9 +305,10 @@ def evaluate(self):
                     # pull a state within that list
                     new_state = random.choice(self.states[new_state_key])
                     # TODO: Fill in more information about the new state
-                    print(
-                        f"\t {key}:\n\t\t{self.event_tracker[key]} -> {self.event_tracker[new_state_key]}\n\t\t{new_state_key}"
-                    )
+                    print(f"\tOld events count: {len(key)}")
+                    print(f"\tOld events: {key}")
+                    print(f"\tNew events count: {len(new_state_key)}")
+                    print(f"\tNew events: {new_state_key}")
                     self.env_recv_queues[key].put(new_state)
                     waiting_for.append(key)
                     # Now copy the hidden state over