class_docstr: An observation visualization wrapper allows to apply custom observation visualization functions
  which are then shown in Tensorboard.
  
  :param env: The environment to wrap.
  :param plot_function: The custom matplotlib plotting function.
class_docstr_tok: ["An","observation","visualization","wrapper","allows","to","apply","custom","observation","visualization","functions","which","are","then","shown","in","Tensorboard",".",":","param","env",":","The","environment","to","wrap",".",":","param","plot_function",":","The","custom","matplotlib","plotting","function","."]
docstr: Keep both actions and observation the same.
docstr_tok: ["Keep","both","actions","and","observation","the","same","."]
code: def get_observation_and_action_dicts(
      self,
      maze_state: Optional[MazeStateType],
      maze_action: Optional[MazeActionType],
      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."""
      return self.env.get_observation_and_action_dicts(
          maze_state, maze_action, first_step_in_episode
      )
  
imports: from abc import ABC
  import warnings
  from typing import TypeVar, Union, Any, Tuple, Dict, Optional, List, Callable
  import matplotlib.pyplot
  import numpy
  from maze.core.annotations import override, unused
  from maze.core.env.action_conversion import ActionType
  from maze.core.env.base_env import BaseEnv
  from maze.core.env.maze_action import MazeActionType
  from maze.core.env.maze_env import MazeEnv
  from maze.core.env.maze_state import MazeStateType
  from maze.core.env.observation_conversion import ObservationType
  from maze.core.env.simulated_env_mixin import SimulatedEnvMixin
  from maze.core.env.structured_env import StructuredEnv
  from maze.core.log_stats.event_decorators import define_plot, define_epoch_stats, define_episode_stats, define_step_stats, define_stats_grouping
  from maze.core.log_stats.reducer_functions import histogram
  from maze.core.utils.factory import Factory
  from maze.core.wrappers.wrapper import Wrapper
label_desc: Use image node_id 3 for calling the ObservationVisualizationWrapper obj's underlying member method code with example usage: obj.get_observation_and_action_dicts(maze_state, maze_action, first_step_in_episode) and returns: self