code: def create_dataset_with_expertise(
      game,
      dataset_expertise=1.0,
      dataset_policies=40,
      dataset_episodes=10000,
      num_datasets=5,
  ):
      """Create offline dataset with varying expertise."""
      idxs, aux_outputs = choose_indices(
          game,
          dataset_expertise=dataset_expertise,
          dataset_policies=dataset_policies,
          dataset_episodes=dataset_episodes,
          num_datasets=num_datasets,
      )
      dataset = _get_dataset_with_idxs(
          game, idxs, num_datasets=num_datasets
      )
      return dataset, aux_outputs
  
file_path: dopamine/dopamine/labs/offline_rl/rlu_tfds/scaling_dataset_utils.py
imports: import concurrent.futures
  from dopamine.labs.offline_rl.rlu_tfds import tfds_atari_utils
  import gin
  import numpy
  import tensorflow_datasets
label_desc: Use image node_id 4 for calling a global function with example usage: create_dataset_with_expertise(game, dataset_expertise, dataset_policies, dataset_episodes, num_datasets) and returns: dataset,  aux_outputs