Undo change in predict_utils

keras-team · Sep 1, 2023 · a175ddb · a175ddb
1 parent 4b61fa4
commit a175ddb
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/keras_cv/models/object_detection/predict_utils.py b/keras_cv/models/object_detection/predict_utils.py
@@ -15,13 +15,13 @@
 import tensorflow as tf
 
 try:
-    from tf.keras.src.engine.training import _minimum_control_deps
-    from tf.keras.src.engine.training import reduce_per_replica
-    from tf.keras.src.utils import tf_utils
+    from keras.src.engine.training import _minimum_control_deps
+    from keras.src.engine.training import reduce_per_replica
+    from keras.src.utils import tf_utils
 except ImportError:
-    from tf.keras.engine.training import _minimum_control_deps
-    from tf.keras.engine.training import reduce_per_replica
-    from tf.keras.utils import tf_utils
+    from keras.engine.training import _minimum_control_deps
+    from keras.engine.training import reduce_per_replica
+    from keras.utils import tf_utils
 
 
 def make_predict_function(model, force=False):