def METHOD_NAME(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)