Pin `ray` `numpy<=2.0.0` test (#17245)

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
pull/17253/head
Laughing 1 month ago committed by GitHub
parent 886d0c7127
commit 83404afff1
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
  1. 3
      ultralytics/utils/tuner.py

@ -1,6 +1,5 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license
import subprocess
from ultralytics.cfg import TASK2DATA, TASK2METRIC, get_save_dir
from ultralytics.utils import DEFAULT_CFG, DEFAULT_CFG_DICT, LOGGER, NUM_THREADS, checks
@ -39,7 +38,7 @@ def run_ray_tune(
train_args = {}
try:
subprocess.run("pip install ray[tune]".split(), check=True) # do not add single quotes here
checks.check_requirements(("ray[tune]", "numpy<2.0.0"))
import ray
from ray import tune

Loading…
Cancel
Save