code: def _generate_split(num_examples: int) -> tf.data.Dataset:
      data = {
          "a": list(range(num_examples)),
          "b": list(range(num_examples)),
      }
      return tf.data.Dataset.from_tensor_slices(data)
  
imports: import tempfile
  from absl.testing import absltest
  from absl.testing import parameterized
  import tensorflow
  import tensorflow_datasets
  from tensorflow_datasets.core import dataset_info
  from tensorflow_datasets.core import sequential_writer
label_desc: Use image node_id 1 for calling a global function with example usage: _generate_split(num_examples) and returns: tf