Internal change

grasskin · edward-bot · commit af2ab9cdb492 · 2023-08-24T07:10:16.000-07:00
PiperOrigin-RevId: 559730941
diff --git a/experimental/marginalization_mixup/augment.py b/experimental/marginalization_mixup/augment.py
@@ -17,7 +17,7 @@
 
 import math
 
-from keras.layers.preprocessing import image_preprocessing as image_ops
+from tensorflow.keras.layers.preprocessing import image_preprocessing as image_ops
 import tensorflow as tf
 
 
