From 49042a4dbbe7d7f04153b7f4088b3fbfe4fc53e4 Mon Sep 17 00:00:00 2001 From: thatguy11325 <148832074+thatguy11325@users.noreply.github.com> Date: Mon, 8 Jul 2024 11:52:31 -0400 Subject: [PATCH] even more --- pokemonred_puffer/environment.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/pokemonred_puffer/environment.py b/pokemonred_puffer/environment.py index a91f210..e716b68 100644 --- a/pokemonred_puffer/environment.py +++ b/pokemonred_puffer/environment.py @@ -294,6 +294,9 @@ def reset(self, seed: Optional[int] = None, options: Optional[dict[str, Any]] = ) # A bit of duplicate code. Blah. self.events = EventFlags(self.pyboy) + self.missables = MissableFlags(self.pyboy) + self.wd728 = Wd728Flags(self.pyboy) + self.party = PartyMons(self.pyboy) self.required_events = self.get_required_events() self.seen_pokemon = np.zeros(152, dtype=np.uint8) self.caught_pokemon = np.zeros(152, dtype=np.uint8)