code: def main(argv):
      del argv
      dataset = benchmark_utils.get_dataset(
          FLAGS.dataset, base_dir=FLAGS.output_base_dir
      )
  
      # Regenerate the dataset and models.
      logging.info("Using dataset: %s", FLAGS.dataset)
      logging.info("Generating raw dataset")
      dataset.generate_raw_dataset(args=FLAGS.generate_dataset_args)
      logging.info("Generating models")
      logging.info("force_tf_compat_v1: %s", FLAGS.force_tf_compat_v1)
      dataset.generate_models(
          args=FLAGS.generate_dataset_args,
          force_tf_compat_v1=FLAGS.force_tf_compat_v1,
      )
  
      # Regenerate intermediate outputs for TFT benchmarks.
      logging.info("Generating intermediate outputs for TFT benchmarks")
      tft_benchmark_base.regenerate_intermediates_for_dataset(
          dataset, force_tf_compat_v1=FLAGS.force_tf_compat_v1
      )
  
imports: from absl import app
  from absl import flags
  from absl import logging
  from tfx.benchmarks import benchmark_utils
  from tfx.benchmarks import tft_benchmark_base
label_desc: Use image node_id 1 for calling a global function with example usage: main(argv) without return types