WIP

build-docker-ant.sh

@@ -94,7 +94,7 @@ docker run --rm \
     antgroup/base-deps:dev$GPU \
     pip download --no-cache-dir --quiet --no-deps --only-binary=:all: "$ANT_RAY_PACKAGE" -d /wheels
 
-WHEEL_FILE=$(ls "$RAY_BUILD_DIR"/.whl/*.whl 2>/dev/null | head -n1)
+WHEEL_FILE=$(find "$RAY_BUILD_DIR/.whl" -name '*.whl' -type f -print -quit 2>/dev/null)
 if [[ -z "$WHEEL_FILE" ]]; then
     echo "Error: No wheel downloaded for ant-ray" >&2
     exit 1

ci/docker/core.build.ant.Dockerfile

@@ -0,0 +1,16 @@
+ARG DOCKER_IMAGE_BASE
+FROM $DOCKER_IMAGE_BASE
+
+WORKDIR /rayci
+
+RUN <<EOF
+#!/bin/bash
+
+set -euo pipefail
+
+chmod 755 /usr/lib/sudo/sudoers.so
+apt-get update && apt-get install -y --reinstall sudo
+
+BUILD=1 DL=1 ./ci/env/install-dependencies.sh
+
+EOF

ci/ray_ci/builder.py

@@ -84,7 +84,6 @@ def main(
     """
     Build a wheel or jar artifact
     """
-    docker_login(_DOCKER_ECR_REPO.split("/")[0])
     ci_init()
     if artifact_type == "wheel":
         logger.info(f"Building wheel for {python_version}")

ci/ray_ci/linux_container.py

@@ -47,8 +47,8 @@ def install_ray(self, build_type: Optional[str] = None) -> List[str]:
                 "-t",
                 self._get_docker_image(),
                 "-f",
-                "/ray/ci/ray_ci/tests.env.Dockerfile",
-                "/ray",
+                f"{os.environ.get('RAY_DIR')}/ci/ray_ci/tests.env.Dockerfile",
+                os.environ.get("RAY_DIR"),
             ],
             env=env,
             stdout=sys.stdout,
@@ -84,4 +84,10 @@ def get_run_command_extra_args(
         return extra_args
 
     def get_artifact_mount(self) -> Tuple[str, str]:
-        return ("/tmp/artifacts", "/artifact-mount")
+        os.makedirs(
+            os.path.join(os.environ.get("BAZEL_DIR"), "artifacts"), exist_ok=True
+        )
+        return (
+            os.path.join(os.environ.get("BAZEL_DIR"), "artifacts"),
+            os.environ.get("BAZEL_DIR"),
+        )