same type attack bonus

thatguy11325 · Oct 19, 2024 · 94e2460 · 94e2460
1 parent 3a58748
commit 94e2460
Showing 1 changed file with 13 additions and 8 deletions.
diff --git 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,
                 ),