Skip to content

Commit

Permalink
Pin ray numpy<=2.0.0 test (ultralytics#17245)
Browse files Browse the repository at this point in the history
Co-authored-by: Glenn Jocher <[email protected]>
Co-authored-by: UltralyticsAssistant <[email protected]>
  • Loading branch information
3 people authored Oct 29, 2024
1 parent 886d0c7 commit 83404af
Showing 1 changed file with 1 addition and 2 deletions.
3 changes: 1 addition & 2 deletions ultralytics/utils/tuner.py
Original file line number Diff line number Diff line change
@@ -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
Expand Down Expand Up @@ -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
Expand Down

0 comments on commit 83404af

Please sign in to comment.