From ff0cc2c0dee17980afbe12f6dcee0d78893bab85 Mon Sep 17 00:00:00 2001 From: Peter Andreas Entschev Date: Mon, 13 May 2024 02:29:23 -0700 Subject: [PATCH] Reenable raft-dask wheel tests requiring UCX-Py --- ci/test_wheel_raft_dask.sh | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/ci/test_wheel_raft_dask.sh b/ci/test_wheel_raft_dask.sh index b8ca54f6e8..bd531e7e85 100755 --- a/ci/test_wheel_raft_dask.sh +++ b/ci/test_wheel_raft_dask.sh @@ -15,11 +15,11 @@ python -m pip install "raft_dask-${RAPIDS_PY_CUDA_SUFFIX}[test]>=0.0.0a0" --find test_dir="python/raft-dask/raft_dask/test" -# rapids-logger "pytest raft-dask" -# python -m pytest --import-mode=append ${test_dir} +rapids-logger "pytest raft-dask" +python -m pytest --import-mode=append ${test_dir} -# rapids-logger "pytest raft-dask (ucx-py only)" -# python -m pytest --import-mode=append ${test_dir} --run_ucx +rapids-logger "pytest raft-dask (ucx-py only)" +python -m pytest --import-mode=append ${test_dir} --run_ucx rapids-logger "pytest raft-dask (ucxx only)" python -m pytest --import-mode=append ${test_dir} --run_ucxx