Internal change

PiperOrigin-RevId: 401839863

official/nlp/keras_nlp/encoders/bert_encoder.py

@@ -24,9 +24,7 @@ from official.nlp.modeling import networks
 class BertEncoder(networks.BertEncoder):
   """Deprecated."""
 
-  def __init__(
-      self,
-      **kwargs):
+  def __init__(self, *args, **kwargs):
     if 'dict_outputs' in kwargs:
       kwargs.pop('dict_outputs')
-    super().__init__(dict_outputs=True, **kwargs)
+    super().__init__(*args, dict_outputs=True, **kwargs)

official/vision/beta/projects/volumetric_models/configs/backbones.py → official/projects/volumetric_models/configs/backbones.py

@@ -14,9 +14,8 @@
 
 # Lint as: python3
 """Backbones configurations."""
-from typing import Optional, Sequence
-
 import dataclasses
+from typing import Optional, Sequence
 
 from official.modeling import hyperparams
 

official/vision/beta/projects/volumetric_models/configs/decoders.py → official/projects/volumetric_models/configs/decoders.py

@@ -14,9 +14,8 @@
 
 # Lint as: python3
 """Decoders configurations."""
-from typing import Optional, Sequence
-
 import dataclasses
+from typing import Optional, Sequence
 
 from official.modeling import hyperparams
 

official/vision/beta/projects/volumetric_models/configs/semantic_segmentation_3d.py → official/projects/volumetric_models/configs/semantic_segmentation_3d.py

@@ -14,17 +14,16 @@
 
 # Lint as: python3
 """Semantic segmentation configuration definition."""
-from typing import List, Optional, Union
 
 import dataclasses
-
+from typing import List, Optional, Union
 from official.core import exp_factory
 from official.modeling import hyperparams
 from official.modeling import optimization
 from official.modeling.hyperparams import config_definitions as cfg
+from official.projects.volumetric_models.configs import backbones
+from official.projects.volumetric_models.configs import decoders
 from official.vision.beta.configs import common
-from official.vision.beta.projects.volumetric_models.configs import backbones
-from official.vision.beta.projects.volumetric_models.configs import decoders
 
 
 @dataclasses.dataclass

official/vision/beta/projects/volumetric_models/configs/semantic_segmentation_3d_test.py → official/projects/volumetric_models/configs/semantic_segmentation_3d_test.py

@@ -21,7 +21,7 @@ import tensorflow as tf
 
 from official.core import exp_factory
 from official.modeling.hyperparams import config_definitions as cfg
-from official.vision.beta.projects.volumetric_models.configs import semantic_segmentation_3d as exp_cfg
+from official.projects.volumetric_models.configs import semantic_segmentation_3d as exp_cfg
 
 
 class ImageSegmentationConfigTest(tf.test.TestCase, parameterized.TestCase):

official/vision/beta/projects/volumetric_models/dataloaders/segmentation_input_3d_test.py → official/projects/volumetric_models/dataloaders/segmentation_input_3d_test.py

@@ -22,8 +22,8 @@ import tensorflow as tf
 from official.common import dataset_fn
 from official.core import config_definitions as cfg
 from official.core import input_reader
+from official.projects.volumetric_models.dataloaders import segmentation_input_3d
 from official.vision.beta.dataloaders import tfexample_utils
-from official.vision.beta.projects.volumetric_models.dataloaders import segmentation_input_3d
 
 
 class InputReaderTest(parameterized.TestCase, tf.test.TestCase):

official/vision/beta/projects/volumetric_models/evaluation/segmentation_metrics.py → official/projects/volumetric_models/evaluation/segmentation_metrics.py

@@ -16,7 +16,7 @@
 from typing import Optional
 
 import tensorflow as tf
-from official.vision.beta.projects.volumetric_models.losses import segmentation_losses
+from official.projects.volumetric_models.losses import segmentation_losses
 
 
 class DiceScore:

official/vision/beta/projects/volumetric_models/evaluation/segmentation_metrics_test.py → official/projects/volumetric_models/evaluation/segmentation_metrics_test.py

@@ -17,7 +17,7 @@
 from absl.testing import parameterized
 import tensorflow as tf
 
-from official.vision.beta.projects.volumetric_models.evaluation import segmentation_metrics
+from official.projects.volumetric_models.evaluation import segmentation_metrics
 
 
 class SegmentationMetricsTest(parameterized.TestCase, tf.test.TestCase):

official/vision/beta/projects/volumetric_models/losses/segmentation_losses_test.py → official/projects/volumetric_models/losses/segmentation_losses_test.py

@@ -17,7 +17,7 @@
 from absl.testing import parameterized
 import tensorflow as tf
 
-from official.vision.beta.projects.volumetric_models.losses import segmentation_losses
+from official.projects.volumetric_models.losses import segmentation_losses
 
 
 class SegmentationLossDiceScoreTest(parameterized.TestCase, tf.test.TestCase):

official/vision/beta/projects/volumetric_models/modeling/backbones/__init__.py → official/projects/volumetric_models/modeling/backbones/__init__.py

@@ -15,4 +15,4 @@
 # Lint as: python3
 """Backbones package definition."""
 
-from official.vision.beta.projects.volumetric_models.modeling.backbones.unet_3d import UNet3D
+from official.projects.volumetric_models.modeling.backbones.unet_3d import UNet3D

official/vision/beta/projects/volumetric_models/modeling/backbones/unet_3d.py → official/projects/volumetric_models/modeling/backbones/unet_3d.py

@@ -24,8 +24,8 @@ from typing import Any, Mapping, Sequence
 # Import libraries
 import tensorflow as tf
 from official.modeling import hyperparams
+from official.projects.volumetric_models.modeling import nn_blocks_3d
 from official.vision.beta.modeling.backbones import factory
-from official.vision.beta.projects.volumetric_models.modeling import nn_blocks_3d
 
 layers = tf.keras.layers
 

official/vision/beta/projects/volumetric_models/modeling/backbones/unet_3d_test.py → official/projects/volumetric_models/modeling/backbones/unet_3d_test.py

@@ -19,7 +19,7 @@
 from absl.testing import parameterized
 import tensorflow as tf
 
-from official.vision.beta.projects.volumetric_models.modeling.backbones import unet_3d
+from official.projects.volumetric_models.modeling.backbones import unet_3d
 
 
 class UNet3DTest(parameterized.TestCase, tf.test.TestCase):

official/vision/beta/projects/volumetric_models/modeling/decoders/__init__.py → official/projects/volumetric_models/modeling/decoders/__init__.py

@@ -15,4 +15,4 @@
 # Lint as: python3
 """Decoders package definition."""
 
-from official.vision.beta.projects.volumetric_models.modeling.decoders.unet_3d_decoder import UNet3DDecoder
+from official.projects.volumetric_models.modeling.decoders.unet_3d_decoder import UNet3DDecoder

official/vision/beta/projects/volumetric_models/modeling/decoders/factory_test.py → official/projects/volumetric_models/modeling/decoders/factory_test.py

@@ -18,10 +18,10 @@ from absl.testing import parameterized
 import tensorflow as tf
 
 from tensorflow.python.distribute import combinations
-from official.vision.beta.projects.volumetric_models.configs import decoders as decoders_cfg
-from official.vision.beta.projects.volumetric_models.configs import semantic_segmentation_3d as semantic_segmentation_3d_exp
-from official.vision.beta.projects.volumetric_models.modeling import decoders
-from official.vision.beta.projects.volumetric_models.modeling.decoders import factory
+from official.projects.volumetric_models.configs import decoders as decoders_cfg
+from official.projects.volumetric_models.configs import semantic_segmentation_3d as semantic_segmentation_3d_exp
+from official.projects.volumetric_models.modeling import decoders
+from official.projects.volumetric_models.modeling.decoders import factory
 
 
 class FactoryTest(tf.test.TestCase, parameterized.TestCase):

official/vision/beta/projects/volumetric_models/modeling/decoders/unet_3d_decoder.py → official/projects/volumetric_models/modeling/decoders/unet_3d_decoder.py

@@ -24,8 +24,8 @@ from typing import Any, Dict, Mapping, Optional, Sequence
 import tensorflow as tf
 
 from official.modeling import hyperparams
-from official.vision.beta.projects.volumetric_models.modeling import nn_blocks_3d
-from official.vision.beta.projects.volumetric_models.modeling.decoders import factory
+from official.projects.volumetric_models.modeling import nn_blocks_3d
+from official.projects.volumetric_models.modeling.decoders import factory
 
 layers = tf.keras.layers
 

official/vision/beta/projects/volumetric_models/modeling/decoders/unet_3d_decoder_test.py → official/projects/volumetric_models/modeling/decoders/unet_3d_decoder_test.py

@@ -19,8 +19,8 @@
 from absl.testing import parameterized
 import tensorflow as tf
 
-from official.vision.beta.projects.volumetric_models.modeling.backbones import unet_3d
-from official.vision.beta.projects.volumetric_models.modeling.decoders import unet_3d_decoder
+from official.projects.volumetric_models.modeling.backbones import unet_3d
+from official.projects.volumetric_models.modeling.decoders import unet_3d_decoder
 
 
 class UNet3DDecoderTest(parameterized.TestCase, tf.test.TestCase):