fix simpler bag obs

thatguy11325 · Jun 16, 2024 · 37cfe60 · 37cfe60
1 parent 289f539
commit 37cfe60
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/pokemonred_puffer/environment.py b/pokemonred_puffer/environment.py
@@ -477,7 +477,8 @@ def _get_obs(self):
         # player_x, player_y, map_n = self.get_game_coords()
         _, wBagItems = self.pyboy.symbol_lookup("wBagItems")
         bag = np.array(self.pyboy.memory[wBagItems : wBagItems + 40])
-        numBagItems = self.pyboy.symbol_lookup("wNumBagItems")
+        numBagItems = self.read_m("wNumBagItems")
+        # item ids start at 1 so using 0 as the nothing value is okay
         bag[2 * numBagItems :] = 0
 
         return {