changed a way of loading proto file for Waymo dataset.

PiperOrigin-RevId: 303070221

Author: cyfra

tensorflow_datasets/object_detection/waymo_open_dataset.py

@@ -23,7 +23,7 @@
 import os
 from absl import logging
 import tensorflow.compat.v2 as tf
-import tensorflow_datasets.proto.waymo_dataset_pb2 as open_dataset
+from tensorflow_datasets.proto import waymo_dataset_pb2 as open_dataset
 import tensorflow_datasets.public_api as tfds
 
 _CITATION = """

tensorflow_datasets/proto/__init__.py

@@ -13,3 +13,16 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+# Lint as: python3
+"""Public API of the proto package."""
+# pylint: disable=g-import-not-at-top,g-importing-member, import-outside-toplevel
+
+
+def _get():
+  from tensorflow_datasets.proto import waymo_dataset_generated_pb2 as waymo_dataset_pb2_
+  return waymo_dataset_pb2_
+
+
+waymo_dataset_pb2 = _get()  # pylint: disable=invalid-name
+del _get
+