Use more specific import tensorflow_docs.api_generator

PiperOrigin-RevId: 504677663

Author: tensorflower-gardener

tools/templates/build_docs.py

@@ -25,7 +25,7 @@
 from absl import app
 from absl import flags
 
-import tensorflow_docs
+import tensorflow_docs.api_generator
 from tensorflow_docs.api_generator import generate_lib
 from tensorflow_docs.api_generator import public_api
 
@@ -67,10 +67,10 @@ def gen_api_docs():
 
   doc_generator = generate_lib.DocGenerator(
       root_title=PROJECT_FULL_NAME,
-      # Replace `tensorflow_docs` with your module, here.
-      py_modules=[(PROJECT_SHORT_NAME, tensorflow_docs)],
-      # Replace `tensorflow_docs` with your module, here.
-      base_dir=os.path.dirname(tensorflow_docs.__file__),
+      # Replace `tensorflow_docs.api_generator` with your module, here.
+      py_modules=[(PROJECT_SHORT_NAME, tensorflow_docs.api_generator)],
+      # Replace `tensorflow_docs.api_generator` with your module, here.
+      base_dir=os.path.dirname(tensorflow_docs.api_generator.__file__),
       code_url_prefix=_URL_PREFIX.value,
       search_hints=_SEARCH_HINTS.value,
       site_path=_SITE_PATH.value,

tools/tensorflow_docs/__init__.py

@@ -13,9 +13,3 @@
 # limitations under the License.
 # ==============================================================================
 """tensorflow_docs is a package for generating python api-reference docs."""
-
-from __future__ import absolute_import
-from __future__ import division
-from __future__ import print_function
-
-from tensorflow_docs import api_generator