diff --git a/tests/test_exports.py b/tests/test_exports.py index 7bc14c1c78..98e4049d79 100644 --- a/tests/test_exports.py +++ b/tests/test_exports.py @@ -32,7 +32,6 @@ def test_export_onnx(): YOLO(file)(SOURCE, imgsz=32) # exported model inference -@pytest.mark.skipif(checks.IS_PYTHON_3_12, reason="OpenVINO not supported in Python 3.12") @pytest.mark.skipif(not TORCH_1_13, reason="OpenVINO requires torch>=1.13") def test_export_openvino(): """Test YOLO exports to OpenVINO format for model inference compatibility.""" diff --git a/ultralytics/utils/benchmarks.py b/ultralytics/utils/benchmarks.py index 4e8fb032b5..8d760f8002 100644 --- a/ultralytics/utils/benchmarks.py +++ b/ultralytics/utils/benchmarks.py @@ -97,8 +97,8 @@ def benchmark( assert MACOS or LINUX, "CoreML and TF.js export only supported on macOS and Linux" assert not IS_RASPBERRYPI, "CoreML and TF.js export not supported on Raspberry Pi" assert not IS_JETSON, "CoreML and TF.js export not supported on NVIDIA Jetson" - if i in {3, 5}: # CoreML and OpenVINO - assert not IS_PYTHON_3_12, "CoreML and OpenVINO not supported on Python 3.12" + if i in {5}: # CoreML + assert not IS_PYTHON_3_12, "CoreML not supported on Python 3.12" if i in {6, 7, 8}: # TF SavedModel, TF GraphDef, and TFLite assert not isinstance(model, YOLOWorld), "YOLOWorldv2 TensorFlow exports not supported by onnx2tf yet" if i in {9, 10}: # TF EdgeTPU and TF.js