pantheonrl.common.observation.extract_partial_obs

extract_partial_obs(observation)[source]

Extract the (partial) observation and the action mask. Used for MAPPO agents.

Parameters:: observation (Observation) – the full observation
Returns:: tuple of (partial) observation and action mask
Return type:: Tuple[ndarray, ndarray | None]