From b3446f58f941c9ff790470317be4a967fcf86986 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Rapha=C3=ABl=20Bournhonesque?= <raphael@bournhonesque.eu>
Date: Tue, 13 Feb 2024 14:36:02 +0100
Subject: [PATCH] fix: fix mypy error in unit test

---
 tests/integration/insights/test_extraction.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/tests/integration/insights/test_extraction.py b/tests/integration/insights/test_extraction.py
index d660172b30..b315469989 100644
--- 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