ObservationWrapper

class maze.core.wrappers.wrapper.ObservationWrapper(env: EnvType)

A Wrapper with typing support modifying the environments observation.

get_observation_and_action_dicts(maze_state: Any | None, maze_action: Any | None, first_step_in_episode: bool) → Tuple[Dict[int | str, Any] | None, Dict[int | str, Any] | None]

(overrides Wrapper)

Convert the observations, keep actions the same.

abstract observation(observation: Any) → Any

Observation mapping method.

reset() → Any

(overrides BaseEnv)

Intercept BaseEnv.reset and map observation.

step(action) → Tuple[Any, Any, bool, Dict[Any, Any]]

Intercept BaseEnv.step and map observation.