Update Tests for coremltools==7.2 (#10274)

ultralytics · Apr 23, 2024 · c0dd296 · c0dd296
1 parent 5a82b51
commit c0dd296
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/test_python.py b/tests/test_python.py
@@ -222,6 +222,7 @@ def test_export_openvino():
     YOLO(f)(SOURCE)  # exported model inference
 
 
+@pytest.mark.skipif(not TORCH_1_9, reason="CoreML>=7.2 not supported with PyTorch<=1.8")
 @pytest.mark.skipif(WINDOWS, reason="CoreML not supported on Windows")  # RuntimeError: BlobWriter not loaded
 @pytest.mark.skipif(IS_RASPBERRYPI, reason="CoreML not supported on Raspberry Pi")
 @pytest.mark.skipif(checks.IS_PYTHON_3_12, reason="CoreML not supported in Python 3.12")