[REVIEW] Reenable raft-dask wheel tests requiring UCX-Py (#2307)

With rapidsai/ucx-py#1041 merged, UCX wheels are now fixed and thus reenabling raft-dask wheel tests that require UCX-Py should be safe.

Authors:
  - Peter Andreas Entschev (https://github.com/pentschev)

Approvers:
  - James Lamb (https://github.com/jameslamb)
  - Jake Awe (https://github.com/AyodeAwe)

URL: #2307

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