ObservationWrapper

class maze.core.wrappers.wrapper.ObservationWrapper(*args, **kwds)

A Wrapper with typing support modifying the environments observation.

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

(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.