From a2f2098d6055267e0009dcf39ef0a03f4cf0acbe Mon Sep 17 00:00:00 2001
From: thatguy11325 <148832074+thatguy11325@users.noreply.github.com>
Date: Thu, 27 Jun 2024 13:52:20 -0400
Subject: [PATCH] upload the global map as jpg

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

diff --git a/pokemonred_puffer/cleanrl_puffer.py b/pokemonred_puffer/cleanrl_puffer.py
index aa6e3fe..c8230a3 100644
--- a/pokemonred_puffer/cleanrl_puffer.py
+++ b/pokemonred_puffer/cleanrl_puffer.py
@@ -309,7 +309,9 @@ def evaluate(self):
                             np.stack(self.infos["pokemon_exploration_map"], axis=0)
                         )
                         if self.wandb_client is not None:
-                            self.stats["Media/aggregate_exploration_map"] = wandb.Image(overlay)
+                            self.stats["Media/aggregate_exploration_map"] = wandb.Image(
+                                overlay, file_type="jpg"
+                            )
                     elif "state" in k:
                         continue