code: def main(args):
      del args
  
      input_tfxio = _make_tfxio()
  
      # User-Defined Processing Pipeline
      with beam.Pipeline() as pipeline:
          with tft_beam.Context(temp_dir=tempfile.mkdtemp()):
              raw_dataset = (
                  pipeline
                  | "ReadRecordBatch"
                  >> input_tfxio.BeamSource(batch_size=5),
                  input_tfxio.TensorAdapterConfig(),
              )
              (transformed_data, _), _ = (
                  raw_dataset
                  | "AnalyzeAndTransform"
                  >> tft_beam.AnalyzeAndTransformDataset(
                      _preprocessing_fn, output_record_batches=True
                  )
              )
              transformed_data = (
                  transformed_data | "ExtractRecordBatch" >> beam.Keys()
              )
              _ = transformed_data | "PrintTransformedData" >> beam.Map(
                  _print_record_batch
              )
  
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 4 for calling a global function with example usage: main(args) without return types