Unverified Commit cbdb2522 authored by Rui Qiao's avatar Rui Qiao Committed by GitHub
Browse files

[Misc] Limit to ray[adag] 2.35 to avoid backward incompatible change (#8509)


Signed-off-by: default avatarRui Qiao <ruisearch42@gmail.com>
parent 99aa4edd
...@@ -14,7 +14,7 @@ librosa # required for audio test ...@@ -14,7 +14,7 @@ librosa # required for audio test
opencv-python # required for video test opencv-python # required for video test
peft peft
requests requests
ray[adag]>=2.35 ray[adag]==2.35
sentence-transformers # required for embedding sentence-transformers # required for embedding
soundfile # required for audio test soundfile # required for audio test
compressed-tensors==0.4.0 # required for compressed-tensors compressed-tensors==0.4.0 # required for compressed-tensors
......
...@@ -437,8 +437,10 @@ class RayGPUExecutor(DistributedGPUExecutor): ...@@ -437,8 +437,10 @@ class RayGPUExecutor(DistributedGPUExecutor):
required_version = version.parse("2.35") required_version = version.parse("2.35")
current_version = version.parse( current_version = version.parse(
pkg_resources.get_distribution("ray").version) pkg_resources.get_distribution("ray").version)
if current_version < required_version: # TODO: update the constraint once we adapt to the backward
raise ValueError(f"Ray version {required_version} or greater is " # 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}") f"required, but found {current_version}")
import importlib.util import importlib.util
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment