Reenable raft-dask wheel tests requiring UCX-Py

rapidsai · May 13, 2024 · 59b7623 · 59b7623
1 parent 5fa4f3c
commit 59b7623
Showing 1 changed file with 4 additions and 4 deletions.
diff --git 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