ObservationVisualizationWrapper¶
-
class
maze.core.wrappers.observation_visualization_wrapper.
ObservationVisualizationWrapper
(*args, **kwds)¶ An observation visualization wrapper allows to apply custom observation visualization functions which are then shown in Tensorboard.
- Parameters
env – The environment to wrap.
plot_function – The custom matplotlib plotting function.
-
clone_from
(env: maze.core.wrappers.observation_visualization_wrapper.ObservationVisualizationWrapper) → None¶ (overrides
SimulatedEnvMixin
)implementation of
SimulatedEnvMixin
.
-
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
)Keep both actions and observation the same.
-
step
(action: Dict[str, Union[int, numpy.ndarray]]) → Tuple[Dict[str, numpy.ndarray], float, bool, Dict[Any, Any]]¶ (overrides
BaseEnv
)Triggers logging events for observations, actions and reward.