code: def _make_tfxio() -> dataset_tfxio.DatasetTFXIO:
      """Make DatasetTFXIO."""
      num_elements = 9
      batch_size = 2
      dataset = tf.data.Dataset.range(num_elements).batch(batch_size)
  
      return dataset_tfxio.DatasetTFXIO(dataset=dataset)
  
file_path: transform/examples/dataset_tfxio_example.py
imports: import pprint
  import tempfile
  from absl import app
  import apache_beam
  import tensorflow
  import tensorflow_transform
  import tensorflow_transform.beam.impl
  from tfx_bsl.tfxio import dataset_tfxio
label_desc: Use image node_id 3 for calling a global function with example usage: _make_tfxio() and returns: dataset_tfxio