Skip to content

Commit

Permalink
Use oldest test dependencies from dependencies.yaml in wheel tests
Browse files Browse the repository at this point in the history
  • Loading branch information
seberg committed Jul 16, 2024
1 parent 8d2ec68 commit 78f17bf
Showing 1 changed file with 13 additions and 2 deletions.
15 changes: 13 additions & 2 deletions ci/test_wheel.sh
Original file line number Diff line number Diff line change
Expand Up @@ -6,8 +6,19 @@ set -eou pipefail
RAPIDS_PY_CUDA_SUFFIX="$(rapids-wheel-ctk-name-gen ${RAPIDS_CUDA_VERSION})"
WHEELHOUSE="${PWD}/dist/"
RAPIDS_PY_WHEEL_NAME="rmm_${RAPIDS_PY_CUDA_SUFFIX}" rapids-download-wheels-from-s3 python "${WHEELHOUSE}"
PIP_PACKAGE=$(echo "${WHEELHOUSE}"/rmm_"${RAPIDS_PY_CUDA_SUFFIX}"*.whl | head -n1)

# echo to expand wildcard before adding '[extra]' requires for pip
python -m pip install -v "$(echo "${WHEELHOUSE}"/rmm_${RAPIDS_PY_CUDA_SUFFIX}*.whl)[test]"
# Use `package[test]` to install latest test dependencies or explicitly install oldest.
if [[ $RAPIDS_DEPENDENCIES != "oldest" ]]; then
python -m pip install -v "${PIP_PACKAGE}[test]"
else
rapids-dependency-file-generator \
--output requirements \
--file-key test_python \
--matrix "cuda=${RAPIDS_CUDA_VERSION%.*};arch=$(arch);py=${RAPIDS_PY_VERSION};dependencies=${RAPIDS_DEPENDENCIES}" \
| tee oldest-dependencies.txt

python -m pip install -v "$PIP_PACKAGE" -r oldest-dependencies.txt
fi

python -m pytest ./python/rmm/rmm/tests

0 comments on commit 78f17bf

Please sign in to comment.