fix explore map

thatguy11325 · Jun 23, 2024 · 87ae90b · 87ae90b
1 parent 54c4ef4
commit 87ae90b
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/pokemonred_puffer/environment.py b/pokemonred_puffer/environment.py
@@ -1063,7 +1063,8 @@ def disable_wild_encounter_hook(self, *args, **kwargs):
     def agent_stats(self, action):
         levels = [self.read_m(f"wPartyMon{i+1}Level") for i in range(self.read_m("wPartyCount"))]
         badges = self.read_m("wObtainedBadges")
-        explore_map = self.explore_map[self.explore_map > 0] = 1
+        explore_map = self.explore_map[self.explore_map]
+        explore_map[explore_map > 0] = 1
         return {
             "stats": {
                 "step": self.step_count + self.reset_count * self.max_steps,