Re-enable Dask wheel tests

rapidsai · May 14, 2024 · 6be59e8 · 6be59e8
1 parent 364dcb6
commit 6be59e8
Showing 1 changed file with 4 additions and 7 deletions.
diff --git a/ci/run_cuml_dask_pytests.sh b/ci/run_cuml_dask_pytests.sh
@@ -4,14 +4,11 @@
 # Support invoking run_cuml_dask_pytests.sh outside the script directory
 cd "$(dirname "$(realpath "${BASH_SOURCE[0]}")")"/../python/cuml/tests/dask
 
-# Disable non-UCXX tests for wheels tests
-if [[ -v CONDA_PREFIX ]]; then
-    rapids-logger "pytest cuml-dask (No UCX-Py/UCXX)"
-    python -m pytest --cache-clear "$@" .
+rapids-logger "pytest cuml-dask (No UCX-Py/UCXX)"
+python -m pytest --cache-clear "$@" .
 
-    rapids-logger "pytest cuml-dask (UCX-Py only)"
-    python -m pytest --cache-clear --run_ucx "$@" .
-fi
+rapids-logger "pytest cuml-dask (UCX-Py only)"
+python -m pytest --cache-clear --run_ucx "$@" .
 
 rapids-logger "pytest cuml-dask (UCXX only)"
 python -m pytest --cache-clear --run_ucxx "$@" .