message: use the base tensorflow package.
diff: @@ -36,7 +36,7 @@ import textwrap
  from absl import app
  from absl import flags
  -import tensorflow.compat.v2 as tf
  +import tensorflow as tf
  from tensorflow_docs.api_generator import doc_controls
  from tensorflow_docs.api_generator import doc_generator_visitor
  