Update __init__.py

keras-team · May 17, 2022 · 7179ff2 · 7179ff2
1 parent 914e9dd
commit 7179ff2
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/keras_cv/layers/__init__.py b/keras_cv/layers/__init__.py
@@ -25,6 +25,7 @@
 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