From af2ab9cdb49248c41265ad3e8533ca5157534bda Mon Sep 17 00:00:00 2001 From: Gabriel Rasskin Date: Thu, 24 Aug 2023 06:08:04 -0700 Subject: [PATCH] Internal change PiperOrigin-RevId: 559730941 --- experimental/marginalization_mixup/augment.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/experimental/marginalization_mixup/augment.py b/experimental/marginalization_mixup/augment.py index b0d9f7f0..0c7d2110 100644 --- 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