ObservationLoggingWrapper

class maze.core.wrappers.observation_logging_wrapper.ObservationLoggingWrapper(*args, **kwds)

A observation logging wrapper for BaseEnv.

Parameters

env – The environment to wrap.

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]]]

Keep both actions and observation the same.

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

Create the observation logs on every step