code: def generate_example(i: int, split: str):
      return i, {
          "id": i,
          "id_str": f"split={split} id={i}",
      }
  
imports: import os
  from typing import Iterable
  import apache_beam
  from apache_beam.testing.test_pipeline import TestPipeline
  import numpy
  import tensorflow
  from tensorflow_datasets import testing
  from tensorflow_datasets.core import dataset_builder
  from tensorflow_datasets.core import dataset_utils
  from tensorflow_datasets.core.dataset_builders import adhoc_builder
  import tensorflow_datasets.public_api
label_desc: Use image node_id 1 for calling a global function with example usage: generate_example(i, split) and returns: i,  dict