diff --git a/pokemonred_puffer/environment.py b/pokemonred_puffer/environment.py index c7c8ad8..6d9446a 100644 --- 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,