less np.sum abuse

thatguy11325 · Apr 17, 2024 · b6c0982 · b6c0982
1 parent a5379e8
commit b6c0982
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/pokemonred_puffer/rewards/baseline.py b/pokemonred_puffer/rewards/baseline.py
@@ -214,14 +214,14 @@ class RockTunnelReplicationEnv(TeachCutReplicationEnv):
     def get_game_state_reward(self):
         return {
             "level": self.reward_config["level"] * self.get_levels_reward(),
-            "exploration": self.reward_config["exploration"] * np.sum(self.seen_coords.values()),
+            "exploration": self.reward_config["exploration"] * sum(self.seen_coords.values()),
             "taught_cut": self.reward_config["taught_cut"] * int(self.taught_cut),
             "event": self.reward_config["event"] * self.update_max_event_rew(),
             "seen_pokemon": self.reward_config["seen_pokemon"] * np.sum(self.seen_pokemon),
             "caught_pokemon": self.reward_config["caught_pokemon"] * np.sum(self.caught_pokemon),
             "moves_obtained": self.reward_config["moves_obtained"] * np.sum(self.moves_obtained),
-            "cut_coords": self.reward_config["cut_coords"] * np.sum(self.cut_coords.values()),
-            "cut_tiles": self.reward_config["cut_tiles"] * np.sum(self.cut_tiles),
+            "cut_coords": self.reward_config["cut_coords"] * sum(self.cut_coords.values()),
+            "cut_tiles": self.reward_config["cut_tiles"] * sum(self.cut_tiles),
             "start_menu": (
                 self.reward_config["start_menu"] * self.seen_start_menu * int(self.taught_cut)
             ),