docs: def render(self,
              trajectories: Tuple[NonFluents, Fluents, Fluents, Fluents, np.array],
              batch: Optional[int] = None) -> None:
          '''Renders the simulated `trajectories` for the given `batch`.
  
          Args:
              trajectories: NonFluents, states, actions, interms and rewards.
              batch: Number of batches to render.
          '''
          raise NotImplementedError