From 08350ea8cbe589b7e6e2f8a9a197a150c3d5f149 Mon Sep 17 00:00:00 2001 From: thatguy11325 <148832074+thatguy11325@users.noreply.github.com> Date: Wed, 17 Apr 2024 00:53:20 -0400 Subject: [PATCH] Temporary disable of pokecenter reward --- pokemonred_puffer/environment.py | 1 + pokemonred_puffer/rewards/baseline.py | 2 +- 2 files changed, 2 insertions(+), 1 deletion(-) 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"]