diff --git a/pokemonred_puffer/cleanrl_puffer.py b/pokemonred_puffer/cleanrl_puffer.py index ca96b76..9030541 100644 --- a/pokemonred_puffer/cleanrl_puffer.py +++ b/pokemonred_puffer/cleanrl_puffer.py @@ -335,8 +335,7 @@ def evaluate(self): # Need a way not to reset the env id counter for the driver env # Until then env ids are 1-indexed - print(f"\tEvents count: {self.event_tracker[key]} -> {len(new_state_key)}") - print(f"\tNew events: {new_state_key}") + print(f"\tNew events ({len(new_state_key)}): {new_state_key}") for key in self.event_tracker.keys(): new_state = random.choice(self.states[new_state_key])