From 5c9265a4b2a4e043a525e6ad0d43abf4c68045b8 Mon Sep 17 00:00:00 2001 From: inisis Date: Sun, 8 Sep 2024 16:41:45 +0000 Subject: [PATCH] update yolo test --- tests/test_yolo.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_yolo.py b/tests/test_yolo.py index 68df2d7..45de159 100644 --- a/tests/test_yolo.py +++ b/tests/test_yolo.py @@ -46,7 +46,7 @@ def test_yolov10_export_onnx_matrix(self, request, task, dynamic, int8, half, ba "task, dynamic, int8, half, batch", product(["yolov10n"], [False], [False], [False], [1, 2]), ) - def test_yolov10_export_onnx_matrix(self, request, task, dynamic, int8, half, batch): + def test_yolov10_export_tflite_matrix(self, request, task, dynamic, int8, half, batch): """Exports YOLOv10 models to TFLite and tests inference with varying configurations.""" file = YOLO(task).export( format="tflite",