From 878683cf0e926154e56d820ae3c3682f0b8d8ae5 Mon Sep 17 00:00:00 2001 From: Rui Qiao <161574667+ruisearch42@users.noreply.github.com> Date: Tue, 17 Sep 2024 00:06:26 -0700 Subject: [PATCH] [Misc] Limit to ray[adag] 2.35 to avoid backward incompatible change (#8509) Signed-off-by: Rui Qiao --- requirements-test.txt | 2 +- vllm/executor/ray_gpu_executor.py | 6 ++++-- 2 files changed, 5 insertions(+), 3 deletions(-) diff --git a/requirements-test.txt b/requirements-test.txt index 16a883b81ce50..10d463de27be5 100644 --- a/requirements-test.txt +++ b/requirements-test.txt @@ -14,7 +14,7 @@ librosa # required for audio test opencv-python # required for video test peft requests -ray[adag]>=2.35 +ray[adag]==2.35 sentence-transformers # required for embedding soundfile # required for audio test compressed-tensors==0.4.0 # required for compressed-tensors diff --git a/vllm/executor/ray_gpu_executor.py b/vllm/executor/ray_gpu_executor.py index b124fe2e08ea6..9433dce842b09 100644 --- a/vllm/executor/ray_gpu_executor.py +++ b/vllm/executor/ray_gpu_executor.py @@ -437,8 +437,10 @@ def _check_ray_adag_installation(self): required_version = version.parse("2.35") current_version = version.parse( pkg_resources.get_distribution("ray").version) - if current_version < required_version: - raise ValueError(f"Ray version {required_version} or greater is " + # TODO: update the constraint once we adapt to the backward + # incompatible API change from ray 2.36 + if current_version != required_version: + raise ValueError(f"Ray version {required_version} is " f"required, but found {current_version}") import importlib.util