Directly importing backend from keras_core (to see whether that still…

… requires TF)
bhky · Dec 1, 2023 · 43bef6b · 43bef6b
1 parent 1a8944d
commit 43bef6b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/_test.py b/tests/_test.py
@@ -5,7 +5,7 @@
 import unittest
 from typing import Optional, Sequence
 
-import keras_core
+from keras_core import backend as keras_backend
 
 import opennsfw2 as n2
 
@@ -38,7 +38,7 @@ def _assert(
                 self.assertTrue(os.path.exists(paths[i]))
 
     def test_predict_images_yahoo_preprocessing(self) -> None:
-        if keras_core.backend.backend() == "tensorflow":
+        if keras_backend.backend() == "tensorflow":
             grad_cam_paths = OUTPUT_GRAD_CAM_PATHS
         else:
             grad_cam_paths = None