Migrate to absl logging

Author: (Ian Stenbit)

examples/layers/preprocessing/cut_mix_demo.py

@@ -8,7 +8,7 @@
 import tensorflow as tf
 import tensorflow_datasets as tfds
 
-from keras_cv.layers.preprocessing import cut_mix
+from keras_cv.layers import preprocessing
 
 IMG_SIZE = (224, 224)
 BATCH_SIZE = 64
@@ -33,7 +33,7 @@ def main():
         .shuffle(10 * BATCH_SIZE)
         .batch(BATCH_SIZE)
     )
-    cutmix = cut_mix.CutMix(1.0)
+    cutmix = preprocessing.CutMix(1.0)
     train_ds = train_ds.map(cutmix, num_parallel_calls=tf.data.AUTOTUNE)
 
     for images, labels in train_ds.take(1):

examples/layers/preprocessing/mix_up_demo.py

@@ -8,7 +8,7 @@
 import tensorflow as tf
 import tensorflow_datasets as tfds
 
-from keras_cv.layers.preprocessing import mix_up
+from keras_cv.layers import preprocessing
 
 IMG_SIZE = (224, 224)
 BATCH_SIZE = 64
@@ -33,7 +33,7 @@ def main():
         .shuffle(10 * BATCH_SIZE)
         .batch(BATCH_SIZE)
     )
-    mixup = mix_up.MixUp(1.0, alpha=0.8)
+    mixup = preprocessing.MixUp(1.0, alpha=0.8)
     train_ds = train_ds.map(mixup, num_parallel_calls=tf.data.AUTOTUNE)
 
     for images, labels in train_ds.take(1):

keras_cv/layers/preprocessing/cut_mix.py

@@ -13,7 +13,7 @@
 # limitations under the License.
 import tensorflow as tf
 import tensorflow.keras.layers as layers
-from tensorflow.python.platform import tf_logging as logging
+from absl import logging
 
 
 class CutMix(layers.Layer):

keras_cv/layers/preprocessing/mix_up.py

@@ -13,7 +13,7 @@
 # limitations under the License.
 import tensorflow as tf
 import tensorflow.keras.layers as layers
-from tensorflow.python.platform import tf_logging as logging
+from absl import logging
 
 
 class MixUp(layers.Layer):

setup.py

@@ -24,7 +24,7 @@
     author="Keras team",
     author_email="[email protected]",
     license="Apache License 2.0",
-    install_requires=["packaging", "tensorflow"],
+    install_requires=["packaging", "tensorflow", "absl-py"],
     extras_require={
         "tests": ["flake8", "isort", "black", "pytest"],
         "examples": ["tensorflow_datasets", "matplotlib"],