From da1bf9d79d4063eb9f389f9e7509cd936e21a9cc Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Mon, 28 Oct 2024 20:17:35 +0100 Subject: [PATCH] Pin `numpy<=2.0.0` on macOS (#17221) --- pyproject.toml | 3 ++- ultralytics/engine/exporter.py | 1 + 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/pyproject.toml b/pyproject.toml index f6cb23204aa..2545739bab0 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -61,7 +61,8 @@ classifiers = [ # Required dependencies ------------------------------------------------------------------------------------------------ dependencies = [ - "numpy>=1.23.0", # temporary patch for compat errors https://github.com/ultralytics/yolov5/actions/runs/9538130424/job/26286956354 + "numpy>=1.23.0", + "numpy<2.0.0; sys_platform == 'darwin'", # macOS OpenVINO errors https://github.com/ultralytics/ultralytics/pull/17221 "matplotlib>=3.3.0", "opencv-python>=4.6.0", "pillow>=7.1.2", diff --git a/ultralytics/engine/exporter.py b/ultralytics/engine/exporter.py index 6d403a2afbf..49e84af9f51 100644 --- a/ultralytics/engine/exporter.py +++ b/ultralytics/engine/exporter.py @@ -457,6 +457,7 @@ def export_onnx(self, prefix=colorstr("ONNX:")): @try_export def export_openvino(self, prefix=colorstr("OpenVINO:")): """YOLO OpenVINO export.""" + # WARNING: numpy>=2.0.0 issue with OpenVINO on macOS https://github.com/ultralytics/ultralytics/pull/17221 check_requirements(f'openvino{"<=2024.0.0" if ARM64 else ">=2024.0.0"}') # fix OpenVINO issue on ARM64 import openvino as ov