code: def plot_episode_states(
      episode_transitions,
      state_variable: str = "state",
      observation_index_in_stack: int = 0,
  ):
      observations = []
      for transition in episode_transitions:
          observations.append(
              np.array(
                  getattr(transition, state_variable)["observation"]
              )[..., observation_index_in_stack]
          )
      show_observation_stack(observations, False)
  
file_path: coach/rl_coach/debug_utils.py
imports: import math
  import matplotlib.pyplot
  import numpy
  from rl_coach.filters.observation.observation_stacking_filter import LazyStack
label_desc: Use image node_id 4 for calling a global function with example usage: plot_episode_states(episode_transitions, state_variable, observation_index_in_stack) without return types