diff --git a/pokemonred_puffer/environment.py b/pokemonred_puffer/environment.py index 47f3cdd..2bc1e6f 100644 --- a/pokemonred_puffer/environment.py +++ b/pokemonred_puffer/environment.py @@ -620,14 +620,19 @@ def _get_obs(self): "events": np.concatenate( ( np.fromiter(self.events.get_events(EVENTS), dtype=np.uint8), - [ - self.read_m("wSSAnne2FCurScript") == 4, # rival 3 - self.missables.get_missable( - "HS_GAME_CORNER_ROCKET" - ), # game corner rocket - self.flags.get_bit("BIT_GAVE_SAFFRON_GUARDS_DRINK"), # saffron guard - self.flags.get_bit("BIT_GOT_LAPRAS"), # got lapras - ], + np.array( + [ + self.read_m("wSSAnne2FCurScript") == 4, # rival 3 + self.missables.get_missable( + "HS_GAME_CORNER_ROCKET" + ), # game corner rocket + self.flags.get_bit( + "BIT_GAVE_SAFFRON_GUARDS_DRINK" + ), # saffron guard + self.flags.get_bit("BIT_GOT_LAPRAS"), # got lapras + ], + dtype=np.uint8, + ), ), dtype=np.uint8, ),