From d3d340e4fd0b524d239431917deceadbd2f036ac Mon Sep 17 00:00:00 2001 From: thatguy11325 <148832074+thatguy11325@users.noreply.github.com> Date: Mon, 8 Jul 2024 19:19:02 -0400 Subject: [PATCH] more print --- 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 36738f8..cac7777 100644 --- a/pokemonred_puffer/cleanrl_puffer.py +++ b/pokemonred_puffer/cleanrl_puffer.py @@ -291,11 +291,11 @@ def evaluate(self): key=lambda x: x[1][1], ) ] - print("Migrating states:") waiting_for = [] # find the envs not in the largest to_migrate_keys = set(self.event_tracker.keys()) - set(largest) + print(f"Migrating {len(to_migrate_keys)} states:") # Need a way not to reset the env id counter for the driver env # Until then env ids are 1-indexed for key in to_migrate_keys: