code: def main(_):
      for cls in [tf.Module, tf.keras.layers.Layer]:
          doc_controls.decorate_all_class_attributes(
              decorator=doc_controls.do_not_doc_in_subclasses,
              cls=cls,
              skip=["__init__"],
          )
  
      doc_generator = generate_lib.DocGenerator(
          root_title="TF-Agents",
          py_modules=[("tf_agents", tf_agents)],
          base_dir=os.path.dirname(tf_agents.__file__),
          code_url_prefix=FLAGS.code_url_prefix,
          search_hints=FLAGS.search_hints,
          site_path=FLAGS.site_path,
          private_map={},
          callbacks=[public_api.local_definitions_filter],
      )
  
      sys.exit(doc_generator.build(output_dir=FLAGS.output_dir))
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import os
  import sys
  from absl import app
  from absl import flags
  import tensorflow
  from tensorflow_docs.api_generator import doc_controls
  from tensorflow_docs.api_generator import generate_lib
  from tensorflow_docs.api_generator import public_api
  import tf_agents
  from tf_agents import agents
  from tf_agents import distributions
  from tf_agents import drivers
  from tf_agents import environments
  from tf_agents import metrics
  from tf_agents import networks
  from tf_agents import policies
  from tf_agents import replay_buffers
  from tf_agents import specs
  from tf_agents import trajectories
  from tf_agents import utils
label_desc: Use image node_id 1 for calling a global function with example usage: main(_) without return types