import keras_cv fails

First, thank you so much to all the contributors for their tremendous effort in building keras-cv.

Obviously, I would like to try it out, but in colab I could not load keras-cv:

https://colab.research.google.com/drive/1k-IrSJF-DlP_o27ohr7GvYsXliGRifTF?usp=sharing

---------------------------------------------------------------------------

ModuleNotFoundError                       Traceback (most recent call last)

[<ipython-input-6-1d020d0fde65>](https://localhost:8080/#) in <module>()
      1 import tensorflow as tf
----> 2 import keras_cv

1 frames

[/usr/local/lib/python3.7/dist-packages/keras_cv/layers/__init__.py](https://localhost:8080/#) in <module>()
     13 # limitations under the License.
     14 
---> 15 from tensorflow.keras.__internal__.layers import BaseImageAugmentationLayer
     16 from tensorflow.keras.layers import CenterCrop
     17 from tensorflow.keras.layers import RandomBrightness

ModuleNotFoundError: No module named 'tensorflow.keras.__internal__.layers'


---------------------------------------------------------------------------
NOTE: If your import is failing due to a missing package, you can
manually install dependencies using either !pip or !apt.

To view examples of installing some common dependencies, click the
"Open Examples" button below.
---------------------------------------------------------------------------

Above is the error message I am getting. What am I missing? Any suggestions would be appreciated.

Thank you, John

[adhadse]

Run this cell:

%%bash
git clone https://github.com/keras-team/keras-cv.git
mv keras-cv/keras_cv /content/keras_cv

pip uninstall tensorflow -y
pip install tf-nightly==2.10.0.dev20220427

pip uninstall keras -y
pip install keras-nightly==2.10.0.dev2022042807

import keras_cv

The library is currently using a nightly version of TensorFlow and Keras for development. This is what we are using, give it a shot.

[qlzh727]

Ack. Will comment on #440.