code: def _make_deep_sea_dataset(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()
  
imports: from typing import Any, List
  from absl import flags
  from bsuite.environments import deep_sea
  import dm_env
  import numpy
  import tensorflow
  import tree
label_desc: Use image node_id 4 for calling a global function with example usage: _make_deep_sea_dataset(environment) and returns: recorder