From 76f1886cc813deffc2b09a8cf6481ce791473331 Mon Sep 17 00:00:00 2001 From: Gil Forsyth Date: Fri, 13 Dec 2024 16:11:38 -0500 Subject: [PATCH] chore: add test_wheel script --- ci/test_wheel.sh | 12 ++++++++++++ 1 file changed, 12 insertions(+) create mode 100644 ci/test_wheel.sh diff --git a/ci/test_wheel.sh b/ci/test_wheel.sh new file mode 100644 index 00000000..74d03b84 --- /dev/null +++ b/ci/test_wheel.sh @@ -0,0 +1,12 @@ +#!/bin/bash +# Copyright (c) 2023, NVIDIA CORPORATION. + +set -eou pipefail + +RAPIDS_PY_CUDA_SUFFIX="$(rapids-wheel-ctk-name-gen ${RAPIDS_CUDA_VERSION})" +RAPIDS_PY_WHEEL_NAME="dask_cuda-${RAPIDS_PY_CUDA_SUFFIX}" RAPIDS_PY_WHEEL_PURE="1" rapids-download-wheels-from-s3 ./dist + +# echo to expand wildcard before adding `[extra]` requires for pip +python -m pip install $(echo ./dist/dask_cuda*.whl)[test] + +python -m pytest -n 8 ./python/dask_cuda/tests