code: def main(_):
      tfrecord_path = os.path.join(FLAGS.tfrecords_dir, FLAGS.split)
      mapping_function = extract_protos.decode_bytes_multiple
      buffer_size, shuffle, cycle_length = 500, False, 1
      dataset = io.get_dataset(
          tfrecord_path,
          mapping_function,
          buffer_size,
          shuffle,
          cycle_length,
      )
      i = 0
      for d in dataset:
          i += 1
          print(d["name"])
          if i % 100 == 0:
              print(i)
      print("final:", i)
  
imports: import os
  from absl import app
  from absl import flags
  import tensorflow_graphics.projects.points_to_3Dobjects.data_preparation.extract_protos
  import tensorflow_graphics.projects.points_to_3Dobjects.utils.io
label_desc: Use image node_id 2 for calling a global function with example usage: main(_) without return types