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)