code: def get_dataset(
      files,
      features,
      batch_size=1,
      shuffle_buffer_size=0,
      parallelism=os.cpu_count(),
      interleave_parallelism=0,
  ):
      """Generates a tf.data.Dataset from a datasource
  
      Args:
        files: A list of files
        features: Dict of features
        batch_size: (Optional.) Batch size for ATDS dataset
        shuffle_buffer_size: (Optional.) Size of the buffer used for shuffling. See
            tensorflow_io/python/experimental/atds/dataset.py for details.
            If unspecified, data is not shuffled.
        parallelism: (Optional.) Number of threads to use while decoding. Defaults
            to all available cores.
      """
      if interleave_parallelism == 0:
          dataset = ATDSDataset(
              filenames=files,
              batch_size=batch_size,
              features=features,
              shuffle_buffer_size=shuffle_buffer_size,
              num_parallel_calls=parallelism,
          )
      else:
          dataset = tf.data.Dataset.list_files(files)
          dataset = dataset.interleave(
              lambda filename: ATDSDataset(
                  filenames=filename,
                  batch_size=batch_size,
                  features=features,
                  shuffle_buffer_size=shuffle_buffer_size,
                  num_parallel_calls=parallelism,
              ),
              cycle_length=interleave_parallelism,
              num_parallel_calls=interleave_parallelism,
          )
      return dataset.prefetch(1)
  
file_path: io/tests/test_atds_avro/utils/atds_benchmark_utils.py
imports: import avro.schema
  import glob
  import json
  import os
  import tensorflow
  from tests.test_atds_avro.utils.data_source import DataSource
  from tests.test_atds_avro.utils.data_source_registry import SMALL_NUM_RECORDS, get_canonical_name, get_data_source_from_registry
  from tests.test_atds_avro.utils.generator.tensor_generator import IntTensorGenerator, FloatTensorGenerator, WordTensorGenerator, BoolTensorGenerator
  from tests.test_atds_avro.utils.generator.sparse_tensor_generator import IntSparseTensorGenerator, FloatSparseTensorGenerator, WordSparseTensorGenerator, BoolSparseTensorGenerator, get_common_value_dist
  from tests.test_atds_avro.utils.generator.varlen_tensor_generator import IntVarLenTensorGenerator, FloatVarLenTensorGenerator, WordVarLenTensorGenerator, BoolVarLenTensorGenerator
  from tensorflow_io.python.experimental.atds.dataset import ATDSDataset
  from tests.test_atds_avro.utils.atds_writer import ATDSWriter
  from tests.test_atds_avro.utils.benchmark_utils import benchmark_func
label_desc: Use image node_id 2 for calling a global function with example usage: get_dataset(files, features, batch_size, shuffle_buffer_size, parallelism, interleave_parallelism) and returns: dataset