code: def make_dataset(environment: dm_env.Environment, stochastic: bool):
      """Make bsuite demos for the current task."""
  
      if not stochastic:
          assert isinstance(environment, deep_sea.DeepSea)
          return _make_deep_sea_dataset(environment)
      else:
          assert isinstance(environment, deep_sea.DeepSea)
          return _make_deep_sea_stochastic_dataset(environment)
  
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 6 for calling a global function with example usage: make_dataset(environment, stochastic) and returns: _make_deep_sea_dataset, _make_deep_sea_stochastic_dataset