code: def main(unused_argv):
      doc_generator = generate_lib.DocGenerator(
          root_title="TensorFlow Model Optimization",
          py_modules=[("tfmot", tfmot)],
          base_dir=os.path.dirname(tfmot.__file__),
          code_url_prefix=FLAGS.code_url_prefix,
          search_hints=FLAGS.search_hints,
          site_path=FLAGS.site_path,
          # TODO(tfmot): remove this once the next release after 0.3.0 happens.
          # This is needed in the interim because the API docs reflect
          # the latest release and the current release still wildcard imports
          # all of the classes below.
          private_map={
              "tfmot.sparsity.keras": [
                  # List of internal classes which get exposed when imported.
                  "InputLayer",
                  "custom_object_scope",
                  "pruning_sched",
                  "pruning_wrapper",
                  "absolute_import",
                  "division",
                  "print_function",
                  "compat",
              ]
          },
      )
  
      doc_generator.build(output_dir=FLAGS.output_dir)
  
file_path: model-optimization/tensorflow_model_optimization/g3doc/tools/build_docs.py
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import os
  from absl import app
  from absl import flags
  from tensorflow_docs.api_generator import generate_lib
  import tensorflow_model_optimization
label_desc: Use image node_id 1 for calling a global function with example usage: main(unused_argv) without return types