def METHOD_NAME(environment: deep_sea.DeepSea):
  """Make DeepSea demonstration dataset."""
  recorder = DemonstrationRecorder()
  _run_optimal_deep_sea_episode(environment, recorder)
  assert recorder.episode_reward > 0
  recorder.record_episode()
  return recorder.make_tf_dataset()