Internal change

PiperOrigin-RevId: 493129643
This commit is contained in:
Mark McDonald 2022-12-05 15:28:52 -08:00 committed by Copybara-Service
parent 3ad03bee0b
commit 99d1dd6fbb

View File

@ -26,7 +26,6 @@ from absl import app
from absl import flags from absl import flags
from tensorflow_docs.api_generator import generate_lib from tensorflow_docs.api_generator import generate_lib
from tensorflow_docs.api_generator import public_api
try: try:
# mediapipe has not been set up to work with bazel yet, so catch & report. # mediapipe has not been set up to work with bazel yet, so catch & report.
@ -68,10 +67,7 @@ def gen_api_docs():
code_url_prefix=_URL_PREFIX.value, code_url_prefix=_URL_PREFIX.value,
search_hints=_SEARCH_HINTS.value, search_hints=_SEARCH_HINTS.value,
site_path=_SITE_PATH.value, site_path=_SITE_PATH.value,
# This callback ensures that docs are only generated for objects that callbacks=[],
# are explicitly imported in your __init__.py files. There are other
# options but this is a good starting point.
callbacks=[public_api.explicit_package_contents_filter],
) )
doc_generator.build(_OUTPUT_DIR.value) doc_generator.build(_OUTPUT_DIR.value)