code: def generate_data():
      # Generate data if requested.
      data_dir = os.path.expanduser(FLAGS.data_dir)
      tmp_dir = os.path.expanduser(FLAGS.tmp_dir)
      tf.gfile.MakeDirs(data_dir)
      tf.gfile.MakeDirs(tmp_dir)
  
      problem_name = FLAGS.problem
      tf.logging.info("Generating data for %s" % problem_name)
      registry.problem(problem_name).generate_data(data_dir, tmp_dir)
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import contextlib
  import os
  import sys
  from tensor2tensor import models
  from tensor2tensor import problems
  from tensor2tensor.data_generators import problem
  from tensor2tensor.utils import cloud_mlengine
  from tensor2tensor.utils import contrib
  from tensor2tensor.utils import decoding
  from tensor2tensor.utils import flags
  from tensor2tensor.utils import hparams_lib
  from tensor2tensor.utils import mlperf_log
  from tensor2tensor.utils import registry
  from tensor2tensor.utils import trainer_lib
  from tensor2tensor.utils import usr_dir
  import tensorflow.compat.v1
  from tensorflow.compat.v1 import estimator
label_desc: Use image node_id 5 for calling a global function with example usage: generate_data() without return types