Cut event obs

pokemonred_puffer/environment.py

@@ -238,6 +238,7 @@ def __init__(self, env_config: pufferlib.namespace):
                 "direction": spaces.Box(low=0, high=4, shape=(1,), dtype=np.uint8),
                 # "reset_map_id": spaces.Box(low=0, high=0xF7, shape=(1,), dtype=np.uint8),
                 "battle_type": spaces.Box(low=0, high=4, shape=(1,), dtype=np.uint8),
+                "cut_event": spaces.Box(low=0, high=1, shape=(1,), dtype=np.uint8),
                 "cut_in_party": spaces.Box(low=0, high=1, shape=(1,), dtype=np.uint8),
                 # "x": spaces.Box(low=0, high=255, shape=(1,), dtype=np.u`int8),
                 # "y": spaces.Box(low=0, high=255, shape=(1,), dtype=np.uint8),
@@ -547,6 +548,7 @@ def _get_obs(self):
             ),
             # "reset_map_id": np.array(self.read_m("wLastBlackoutMap"), dtype=np.uint8),
             "battle_type": np.array(self.read_m("wIsInBattle") + 1, dtype=np.uint8),
+            "cut_event": np.array(self.read_bit(0xD803, 0), dtype=np.uint8),
             "cut_in_party": np.array(self.check_if_party_has_cut(), dtype=np.uint8),
             # "x": np.array(player_x, dtype=np.uint8),
             # "y": np.array(player_y, dtype=np.uint8),

pokemonred_puffer/policies/multi_convolutional.py

@@ -115,6 +115,7 @@ def encode_observations(self, observations):
                     one_hot(observations["direction"].long(), 4).float().squeeze(1),
                     # one_hot(observations["reset_map_id"].long(), 0xF7).float().squeeze(1),
                     one_hot(observations["battle_type"].long(), 4).float().squeeze(1),
+                    observations["cut_event"].float(),
                     observations["cut_in_party"].float(),
                     # observations["x"].float(),
                     # observations["y"].float(),