fix: fix mypy error in unit test

openfoodfacts · Feb 13, 2024 · b3446f5 · b3446f5
1 parent b2333e9
commit b3446f5
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tests/integration/insights/test_extraction.py b/tests/integration/insights/test_extraction.py
@@ -26,7 +26,10 @@ def __init__(self, raw_result: ObjectDetectionRawResult):
         self.raw_result = raw_result
 
     def detect_from_image(
-        self, image: Image.Image, output_image: bool = False
+        self,
+        image: Image.Image,
+        output_image: bool = False,
+        triton_uri: str | None = None,
     ) -> ObjectDetectionRawResult:
         return self.raw_result