0.2.1

keras-team · May 17, 2022 · fd6f257 · fd6f257
1 parent 7179ff2
commit fd6f257
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/keras_cv/__init__.py b/keras_cv/__init__.py
@@ -23,4 +23,4 @@
 
 version_check.check_tf_version()
 
-__version__ = "0.2.0dev"
+__version__ = "0.2.1"
diff --git a/keras_cv/layers/__init__.py b/keras_cv/layers/__init__.py
@@ -12,6 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+from tensorflow.keras.__internal__.layers import BaseImageAugmentationLayer
 from tensorflow.keras.layers import CenterCrop
 from tensorflow.keras.layers import RandomBrightness
 from tensorflow.keras.layers import RandomContrast
@@ -25,7 +26,6 @@
 from tensorflow.keras.layers import Rescaling
 from tensorflow.keras.layers import Resizing
 
-from tensorflow.keras.__internal__.layers import BaseImageAugmentationLayer
 from keras_cv.layers.preprocessing.auto_contrast import AutoContrast
 from keras_cv.layers.preprocessing.channel_shuffle import ChannelShuffle
 from keras_cv.layers.preprocessing.cut_mix import CutMix