even more

thatguy11325 · Jul 8, 2024 · 49042a4 · 49042a4
1 parent 34435df
commit 49042a4
Showing 1 changed file with 3 additions and 0 deletions.
diff --git 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)