[Misc] Use ray[adag] dependency instead of cuda

Signed-off-by: Rui Qiao <ruisearch42@gmail.com>
vllm-project · Aug 28, 2024 · 2c311fa · 2c311fa
1 parent c166e7e
commit 2c311fa
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 7 deletions.
diff --git a/requirements-adag.txt b/requirements-adag.txt
diff --git a/requirements-test.txt b/requirements-test.txt
@@ -1,6 +1,3 @@
-# Needed for Ray accelerated DAG tests
--r requirements-adag.txt
-
 # testing
 pytest
 tensorizer>=2.9.0
@@ -15,7 +12,7 @@ einops # required for MPT and qwen-vl
 httpx
 peft
 requests
-ray
+ray[adag]>=2.35
 sentence-transformers # required for embedding
 compressed-tensors==0.4.0 # required for compressed-tensors
 timm # required for internvl test