Automated Code Change

PiperOrigin-RevId: 927193180

Author: runzw

tensorflow_datasets/core/dataset_info_test.py

@@ -23,6 +23,7 @@
 from unittest import mock
 
 from etils import epath
+from google.protobuf import text_format
 import numpy as np
 import pytest
 import tensorflow as tf
@@ -40,8 +41,6 @@
 from tensorflow_datasets.image_classification import mnist
 from tensorflow_datasets.testing.dummy_config_based_datasets.dummy_ds_1 import dummy_ds_1_dataset_builder
 
-from google.protobuf import text_format
-
 
 _TFDS_DIR = utils.tfds_path()
 _DATA_DIR = os.path.join(_TFDS_DIR, "testing", "test_data", "dataset_info")

tensorflow_datasets/core/features/feature.py

@@ -30,6 +30,9 @@
 from absl import logging
 from etils import enp
 from etils import epath
+from google.protobuf import descriptor
+from google.protobuf import json_format
+from google.protobuf import message
 import numpy as np
 from tensorflow_datasets.core import constants
 from tensorflow_datasets.core import utils
@@ -44,10 +47,6 @@
 from tensorflow_datasets.core.utils.lazy_imports_utils import tf_agents
 from tensorflow_datasets.core.utils.lazy_imports_utils import tree
 
-from google.protobuf import descriptor
-from google.protobuf import json_format
-from google.protobuf import message
-
 Json = type_utils.Json
 Shape = type_utils.Shape
 TreeDict = type_utils.TreeDict

tensorflow_datasets/core/folder_dataset/compute_split_utils.py

@@ -26,6 +26,7 @@
 
 from etils import epath
 from etils import etree
+from google.protobuf import json_format
 from tensorflow_datasets.core import file_adapters
 from tensorflow_datasets.core import lazy_imports_lib
 from tensorflow_datasets.core import naming
@@ -34,8 +35,6 @@
 from tensorflow_datasets.core.utils.lazy_imports_utils import apache_beam as beam
 from tensorflow_datasets.core.utils.lazy_imports_utils import array_record_data_source
 
-from google.protobuf import json_format
-
 _SplitFilesDict = Mapping[str, Sequence[naming.FilenameInfo]]
 
 

tensorflow_datasets/core/read_only_builder_test.py

@@ -23,6 +23,7 @@
 
 import dill
 from etils import epath
+from google.protobuf import json_format
 import pytest
 import tensorflow as tf
 from tensorflow_datasets import testing
@@ -40,7 +41,6 @@
 from tensorflow_datasets.core.proto import dataset_info_pb2
 from tensorflow_datasets.core.utils import error_utils
 from tensorflow_datasets.core.utils import file_utils
-from google.protobuf import json_format
 
 _DATA_DIR = epath.Path('path/to')
 _BUILDER_NAME = 'ds0'