duplicate some code for success

thatguy11325 · Jul 8, 2024 · fd05e9e · fd05e9e
1 parent 8794fd5
commit fd05e9e
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/pokemonred_puffer/environment.py b/pokemonred_puffer/environment.py
@@ -292,6 +292,8 @@ def reset(self, seed: Optional[int] = None, options: Optional[dict[str, Any]] =
                     self.read_m(i).bit_count()
                     for i in range(EVENT_FLAGS_START, EVENT_FLAGS_START + EVENTS_FLAGS_LENGTH)
                 )
+                # A bit of duplicate code. Blah.
+                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)
                 self.moves_obtained = np.zeros(0xA5, dtype=np.uint8)