From 14c812b413ba91450d992e31540ec23126ebe0a3 Mon Sep 17 00:00:00 2001 From: Karl-Joan Alesma Date: Tue, 7 Nov 2023 17:48:57 +0200 Subject: [PATCH] Fix tests --- tests/test_yolov8onnx.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/test_yolov8onnx.py b/tests/test_yolov8onnx.py index c1c6776ea..268bb53e3 100644 --- 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