Temporary disable of pokecenter reward

thatguy11325 · Apr 17, 2024 · 08350ea · 08350ea
1 parent e8fc37b
commit 08350ea
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git 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
@@ -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"]