Fix tests

obss · Nov 25, 2023 · 14c812b · 14c812b
1 parent 131bc0b
commit 14c812b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_yolov8onnx.py b/tests/test_yolov8onnx.py
@@ -30,7 +30,7 @@ def test_load_model(self):
         )
 
         # Test setting options for onnxruntime
-        yolov8_onnx_detection_model.load_model(enable_mem_pattern=False)
+        yolov8_onnx_detection_model.load_model({"enable_mem_pattern": False})
 
         self.assertNotEqual(yolov8_onnx_detection_model.model, None)
 
@@ -85,7 +85,7 @@ def test_perform_inference(self):
         best_bbox = boxes[best_box_index]
 
         # Compare
-        desired_bbox = [833, 309, 872, 342]
+        desired_bbox = [603, 239, 629, 259]
         predicted_bbox = best_bbox.tolist()
         margin = 2