fix log loop

thatguy11325 · Jun 21, 2024 · 7963802 · 7963802
1 parent 4c8fffe
commit 7963802
Showing 1 changed file with 1 addition and 6 deletions.
diff --git a/pokemonred_puffer/cleanrl_puffer.py b/pokemonred_puffer/cleanrl_puffer.py
@@ -297,13 +297,8 @@ def evaluate(self):
                         )
                         if self.wandb_client is not None:
                             self.stats["Media/aggregate_exploration_map"] = wandb.Image(overlay)
-
-                for k, v in self.infos.items():
-                    if "_map" in k and self.wandb_client is not None:
-                        self.stats[f"Media/{k}"] = wandb.Image(v[0])
-                        continue
                     elif "state" in k:
-                        pass
+                        continue
 
                     try:  # TODO: Better checks on log data types
                         self.stats[k] = np.mean(v)