code: def build_docs(output_dir):
      """Build api docs for TensorFlow Ranking package."""
      doc_generator = generate_lib.DocGenerator(
          root_title="TensorFlow Ranking",
          py_modules=[("tfr", tfr)],
          base_dir=os.path.dirname(tfr.__file__),
          search_hints=FLAGS.search_hints,
          code_url_prefix=FLAGS.code_url_prefix,
          site_path=FLAGS.site_path,
          callbacks=[public_api.local_definitions_filter],
      )
  
      doc_generator.build(output_dir)
      print("Output docs to: ", FLAGS.output_dir)
  
file_path: ranking/docs/build_docs.py
imports: import os
  import sys
  from absl import app
  from absl import flags
  from tensorflow_docs.api_generator import generate_lib
  from tensorflow_docs.api_generator import public_api
  import tensorflow_ranking
  import tensorflow_ranking.extension.premade
label_desc: Use image node_id 1 for calling a global function with example usage: build_docs(output_dir) without return types