update init

keras-team · Nov 7, 2023 · 42d6798 · 42d6798
1 parent 4883a17
commit 42d6798
Showing 1 changed file with 5 additions and 7 deletions.
diff --git a/keras_cv/backend/__init__.py b/keras_cv/backend/__init__.py
@@ -24,11 +24,12 @@
 
 import types
 
+from keras_cv.backend.config import (
+    detect_if_tensorflow_uses_keras_3,
+    multi_backend,
+)
 from packaging.version import parse
 
-from keras_cv.backend.config import detect_if_tensorflow_uses_keras_3
-from keras_cv.backend.config import multi_backend
-
 # Keys are of the form: "module.where.attr.exists->module.where.to.alias"
 # Value are of the form: ["attr1", "attr2", ...] or
 #                        [("attr1_original_name", "attr1_alias_name")]
@@ -41,10 +42,7 @@
     ],
     "models->saving": ["load_model"],
 }
-
-if detect_if_tensorflow_uses_keras_3():
-    from tensorflow import keras
-elif multi_backend():
+if multi_backend():
     import keras
 
     if not hasattr(keras, "__version__") or parse(keras.__version__) < parse(