[llm] change test base from ml base to test base

it is a smaller image Signed-off-by: Lonnie Liu <[email protected]>
ray-project · Feb 8, 2025 · b9d7704 · b9d7704
1 parent 9b5691b
commit b9d7704
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/.buildkite/llm.rayci.yml b/.buildkite/llm.rayci.yml
@@ -1,7 +1,7 @@
 group: llm tests
 depends_on:
   - forge
-  - oss-ci-base_ml-multipy
+  - oss-ci-base_test-multipy
 steps:
   - name: llmbuild
     wanda: ci/docker/llm.build.wanda.yaml

diff --git a/ci/docker/llm.build.Dockerfile b/ci/docker/llm.build.Dockerfile
@@ -1,6 +1,6 @@
 # syntax=docker/dockerfile:1.3-labs
 
-ARG DOCKER_IMAGE_BASE_BUILD=cr.ray.io/rayproject/oss-ci-base_ml-py3.11
+ARG DOCKER_IMAGE_BASE_BUILD=cr.ray.io/rayproject/oss-ci-base_test-py3.11
 FROM $DOCKER_IMAGE_BASE_BUILD
 
 ARG RAY_CI_JAVA_BUILD=

diff --git a/ci/docker/llm.build.wanda.yaml b/ci/docker/llm.build.wanda.yaml
@@ -1,5 +1,5 @@
 name: "llmbuild"
-froms: ["cr.ray.io/rayproject/oss-ci-base_ml-py3.11"]
+froms: ["cr.ray.io/rayproject/oss-ci-base_test-py3.11"]
 dockerfile: ci/docker/llm.build.Dockerfile
 srcs:
   - python/requirements_compiled_rayllm_test_py311.txt