From 486d4cbe5461c9b8ffe34b5bc3a9c88ec93ab252 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sun, 22 Sep 2024 16:17:05 +0200 Subject: [PATCH] Update OpenVINO CI for Python 3.12 (#16420) --- tests/test_exports.py | 1 - ultralytics/utils/benchmarks.py | 4 ++-- 2 files changed, 2 insertions(+), 3 deletions(-) diff --git a/tests/test_exports.py b/tests/test_exports.py index 7bc14c1c..98e4049d 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 4e8fb032..8d760f80 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