fix test

pyronear · Apr 4, 2024 · 347c230 · 347c230
1 parent 3e8ff47
commit 347c230
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_vision.py b/tests/test_vision.py
@@ -7,7 +7,7 @@ def test_classifier(mock_wildfire_image):
     # Instantiate the ONNX model
     model = Classifier()
     # Check preprocessing
-    out = model.preprocess_image(mock_wildfire_image)
+    out, pad = model.preprocess_image(mock_wildfire_image)
     assert isinstance(out, np.ndarray) and out.dtype == np.float32
     assert out.shape == (1, 3, 384, 640)
     # Check inference