From 20eb5b7df56c56825a7fec8e104411ca15293dda Mon Sep 17 00:00:00 2001
From: thatguy11325 <148832074+thatguy11325@users.noreply.github.com>
Date: Mon, 1 Jul 2024 12:10:28 -0400
Subject: [PATCH] Forgot to set dtype for events

---
 pokemonred_puffer/environment.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/pokemonred_puffer/environment.py b/pokemonred_puffer/environment.py
index f9d7872..cdba343 100644
--- a/pokemonred_puffer/environment.py
+++ b/pokemonred_puffer/environment.py
@@ -549,7 +549,10 @@ def _get_obs(self):
                 "special": np.array([self.party[i].Special for i in range(6)], dtype=np.uint16),
                 "moves": np.array([self.party[i].Moves for i in range(6)], dtype=np.uint8),
             }
-            | {event: np.array(self.events.get_event(event)) for event in REQUIRED_EVENTS}
+            | {
+                event: np.array(self.events.get_event(event), dtype=np.uint8)
+                for event in REQUIRED_EVENTS
+            }
         )
 
     def set_perfect_iv_dvs(self):