Skip to content

Commit

Permalink
fix event agent stats
Browse files Browse the repository at this point in the history
  • Loading branch information
thatguy11325 committed Jun 24, 2024
1 parent 6c69d35 commit 14cc600
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion pokemonred_puffer/environment.py
Original file line number Diff line number Diff line change
Expand Up @@ -1113,7 +1113,7 @@ def agent_stats(self, action):
"pokecenter": np.sum(self.pokecenters),
}
| {f"badge_{i+1}": bool(badges & (1 << i)) for i in range(8)},
"events": {self.events.get_event(event) for event in REQUIRED_EVENTS}
"events": {event: self.events.get_event(event) for event in REQUIRED_EVENTS}
| {"rival3": int(self.read_m(0xD665) == 4)},
"reward": self.get_game_state_reward(),
"reward/reward_sum": sum(self.get_game_state_reward().values()),
Expand Down

0 comments on commit 14cc600

Please sign in to comment.