Update path to the model

luxonis · Jan 16, 2025 · fe49f63 · fe49f63
1 parent 1218c94
commit fe49f63
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_conversion.py b/tests/test_conversion.py
@@ -8,12 +8,12 @@
 
 def test_automatic_version_detection():
     """Test the autodetection of the model version."""
-    assert detect_version("yolov8n.pt") == "yolov8"
+    assert detect_version("tests/yolov8n.pt") == "yolov8"
 
 
 def test_model_conversion():
     """Test the conversion of a model."""
-    exporter = YoloV8Exporter("yolov8n.pt", (416, 416), True)
+    exporter = YoloV8Exporter("tests/yolov8n.pt", (416, 416), True)
     exporter.export_onnx()
     exporter.export_nn_archive()