Fix pb2 imports

PiperOrigin-RevId: 517229050

Author: MarkDaoust

tools/tensorflow_docs/api_generator/report/linter.py

@@ -26,7 +26,7 @@
 from tensorflow_docs.api_generator import get_source
 from tensorflow_docs.api_generator import parser
 from tensorflow_docs.api_generator.pretty_docs import base_page
-from tensorflow_docs.api_generator.report.schema import api_report_generated_pb2 as api_report_pb2
+from tensorflow_docs.api_generator.report.schema import api_report_pb2
 
 
 

tools/tensorflow_docs/api_generator/report/linter_test.py

@@ -28,7 +28,7 @@
 from tensorflow_docs.api_generator import reference_resolver as reference_resolver_lib
 from tensorflow_docs.api_generator.pretty_docs import docs_for_object
 from tensorflow_docs.api_generator.report import utils
-from tensorflow_docs.api_generator.report.schema import api_report_generated_pb2 as api_report_pb2
+from tensorflow_docs.api_generator.report.schema import api_report_pb2
 
 
 class DummyVisitor(object):

tools/tensorflow_docs/api_generator/report/utils.py

@@ -20,7 +20,7 @@
 from tensorflow_docs.api_generator.pretty_docs import function_page
 from tensorflow_docs.api_generator.report import linter
 
-from tensorflow_docs.api_generator.report.schema import api_report_generated_pb2 as api_report_pb2
+from tensorflow_docs.api_generator.report.schema import api_report_pb2
 from google.protobuf import timestamp_pb2