diff --git a/pokemonred_puffer/environment.py b/pokemonred_puffer/environment.py index ac4646f..ac5a305 100644 --- a/pokemonred_puffer/environment.py +++ b/pokemonred_puffer/environment.py @@ -692,6 +692,7 @@ def agent_stats(self, action): "item_count": self.read_m(0xD31D), "reset_count": self.reset_count, "blackout_count": self.blackout_count, + "pokecenter": np.sum(self.pokecenters), }, "reward": self.get_game_state_reward(), "reward/reward_sum": sum(self.get_game_state_reward().values()), diff --git a/pokemonred_puffer/rewards/baseline.py b/pokemonred_puffer/rewards/baseline.py index 725d79b..d0ca57a 100644 --- a/pokemonred_puffer/rewards/baseline.py +++ b/pokemonred_puffer/rewards/baseline.py @@ -232,7 +232,7 @@ def get_game_state_reward(self): self.reward_config["stats_menu"] * self.seen_stats_menu * int(self.taught_cut) ), "bag_menu": self.reward_config["bag_menu"] * self.seen_bag_menu * int(self.taught_cut), - "pokecenter": self.reward_config["pokecenter"] * np.sum(self.pokecenters), + # "pokecenter": self.reward_config["pokecenter"] * np.sum(self.pokecenters), "badges": self.reward_config["badges"] * self.get_badges(), "met_bill": self.reward_config["bill_saved"] * int(self.read_bit(0xD7F1, 0)), "used_cell_separator_on_bill": self.reward_config["bill_saved"]