From af1fc7b31d5d689ad8ad5c910495b3de7761bf08 Mon Sep 17 00:00:00 2001 From: Gil Forsyth Date: Mon, 6 Jan 2025 11:23:49 -0500 Subject: [PATCH] generate test dependencies and install in test_wheel.sh --- ci/test_wheel.sh | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/ci/test_wheel.sh b/ci/test_wheel.sh index cab01055..4aad0d69 100755 --- a/ci/test_wheel.sh +++ b/ci/test_wheel.sh @@ -5,6 +5,16 @@ set -eou pipefail RAPIDS_PY_WHEEL_NAME="dask-cuda" RAPIDS_PY_WHEEL_PURE="1" rapids-download-wheels-from-s3 python ./dist +# Install cuda-suffixed dependencies b/c while `dask-cuda` has no cuda suffix, the test dependencies do +rapids-dependency-file-generator \ + --output requirements \ + --file-key "test_python" \ + --matrix "cuda=${RAPIDS_CUDA_VERSION%.*};arch=$(arch);py=${RAPIDS_PY_VERSION};cuda_suffixed=true" \ +| tee /tmp/requirements-test.txt + +rapids-logger "Installing test dependencies" +python -m pip install -v --prefer-binary -r /tmp/requirements-test.txt + # echo to expand wildcard before adding `[extra]` requires for pip python -m pip install $(echo ./dist/dask_cuda*.whl)