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

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

Hide file tree

Showing 4 changed files with 1 addition and 10 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -37,7 +37,6 @@ WORKDIR /workspace
 
 # install build and runtime dependencies
 COPY requirements-common.txt requirements-common.txt
-COPY requirements-adag.txt requirements-adag.txt
 COPY requirements-cuda.txt requirements-cuda.txt
 RUN --mount=type=cache,target=/root/.cache/pip \
     python3 -m pip install -r requirements-cuda.txt
@@ -69,7 +68,6 @@ COPY setup.py setup.py
 COPY cmake cmake
 COPY CMakeLists.txt CMakeLists.txt
 COPY requirements-common.txt requirements-common.txt
-COPY requirements-adag.txt requirements-adag.txt
 COPY requirements-cuda.txt requirements-cuda.txt
 COPY pyproject.toml pyproject.toml
 COPY vllm vllm

diff --git a/MANIFEST.in b/MANIFEST.in
@@ -1,5 +1,4 @@
 include LICENSE
-include requirements-adag.txt
 include requirements-common.txt
 include requirements-cuda.txt
 include requirements-rocm.txt

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