fix specs

inaturalist · Dec 31, 2024 · 9edc3d4 · 9edc3d4
1 parent fa3062b
commit 9edc3d4
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/tests/conftest.py b/tests/conftest.py
@@ -36,4 +36,5 @@ def inatInferrer(request, mocker):
             os.path.realpath(os.path.dirname(__file__) + "/fixtures/synonyms.csv")
     }
     mocker.patch("tensorflow.keras.models.load_model", return_value=MagicMock())
+    mocker.patch("tensorflow.keras.Model", return_value=MagicMock())
     return InatInferrer(config)
diff --git a/tests/test_vision_inferrer.py b/tests/test_vision_inferrer.py
@@ -6,6 +6,7 @@
 class TestVisionInferrer:
     def test_initialization(self, mocker):
         mocker.patch("tensorflow.keras.models.load_model", return_value=MagicMock())
+        mocker.patch("tensorflow.keras.Model", return_value=MagicMock())
         model_path = "model_path"
         inferrer = VisionInferrer(model_path)
         assert inferrer.model_path == model_path
@@ -16,6 +17,7 @@ def test_initialization(self, mocker):
 
     def test_process_image(self, mocker):
         mocker.patch("tensorflow.keras.models.load_model", return_value=MagicMock())
+        mocker.patch("tensorflow.keras.Model", return_value=MagicMock())
         model_path = "model_path"
         inferrer = VisionInferrer(model_path)
         theimage = "theimage"