Fix Ray 2.7.0 breaking changes (#4964)

This commit is contained in:
Glenn Jocher 2023-09-19 13:22:23 +02:00 committed by GitHub
parent 742ec7fb1d
commit 0cf82f5040
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
5 changed files with 15 additions and 13 deletions

View file

@ -79,7 +79,6 @@ def test_predict_sam():
@pytest.mark.skipif(not CUDA_IS_AVAILABLE, reason='CUDA is not available')
@pytest.mark.skipif(True, reason="RayTune Error pyarrow.lib.ArrowInvalid: URI has empty scheme: './runs/tune'")
def test_model_ray_tune():
with contextlib.suppress(RuntimeError): # RuntimeError may be caused by out-of-memory
YOLO('yolov8n-cls.yaml').tune(use_ray=True,

View file

@ -41,7 +41,7 @@ def test_model_methods():
model.to('cpu')
model.fuse()
model.clear_callback('on_train_start')
model._reset_callbacks()
model.reset_callbacks()
# Model properties
_ = model.names