Internal change

PiperOrigin-RevId: 436802310

official/common/registry_imports.py

@@ -14,7 +14,7 @@
 
 """All necessary imports for registration."""
 # pylint: disable=unused-import
+from official import vision
 from official.nlp import tasks
 from official.nlp.configs import experiment_configs
 from official.utils.testing import mock_task
-from official.vision import beta

official/vision/beta/serving/detection_test.py

@@ -22,8 +22,8 @@ import numpy as np
 from PIL import Image
 import tensorflow as tf
 
-from official.common import registry_imports  # pylint: disable=unused-import
 from official.core import exp_factory
+from official.vision.beta import configs  # pylint: disable=unused-import
 from official.vision.beta.serving import detection
 
 

official/vision/beta/serving/export_tflite_lib_test.py

@@ -19,8 +19,8 @@ from absl.testing import parameterized
 import tensorflow as tf
 
 from tensorflow.python.distribute import combinations
-from official.common import registry_imports  # pylint: disable=unused-import
 from official.core import exp_factory
+from official.vision.beta import configs  # pylint: disable=unused-import
 from official.vision.beta.dataloaders import tfexample_utils
 from official.vision.beta.serving import detection as detection_serving
 from official.vision.beta.serving import export_tflite_lib

official/vision/beta/serving/image_classification_test.py

@@ -22,8 +22,8 @@ import numpy as np
 from PIL import Image
 import tensorflow as tf
 
-from official.common import registry_imports  # pylint: disable=unused-import
 from official.core import exp_factory
+from official.vision.beta import configs  # pylint: disable=unused-import
 from official.vision.beta.serving import image_classification
 
 

official/vision/beta/serving/semantic_segmentation_test.py

@@ -22,8 +22,8 @@ import numpy as np
 from PIL import Image
 import tensorflow as tf
 
-from official.common import registry_imports  # pylint: disable=unused-import
 from official.core import exp_factory
+from official.vision.beta import configs  # pylint: disable=unused-import
 from official.vision.beta.serving import semantic_segmentation
 
 

official/vision/beta/serving/video_classification_test.py

@@ -12,7 +12,6 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-
 # import io
 import os
 import random
@@ -21,8 +20,8 @@ from absl.testing import parameterized
 import numpy as np
 import tensorflow as tf
 
-from official.common import registry_imports  # pylint: disable=unused-import
 from official.core import exp_factory
+from official.vision.beta import configs  # pylint: disable=unused-import
 from official.vision.beta.dataloaders import tfexample_utils
 from official.vision.beta.serving import video_classification