Update __init__.py

Author: LukeWood

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