def get_frame(env_observation, info):
  """Searches for a rendered frame in 'info', fallbacks to the env obs."""
  info_frame = info.get('frame')
  if info_frame is not None:
    return info_frame
  return env_observation