From 7de88314d8d37ff91ac7c0ed7127487fcf0ce5c4 Mon Sep 17 00:00:00 2001
From: Divye Gala <divyegala@gmail.com>
Date: Wed, 18 Sep 2024 18:05:45 -0400
Subject: [PATCH] Reenable `pytest cuml-dask` for CUDA 12.5 wheel CI tests
 (#6051)

Reference issue: https://github.com/rapidsai/cuml/issues/6050

Authors:
  - Divye Gala (https://github.com/divyegala)
  - Ray Douglass (https://github.com/raydouglass)
  - https://github.com/jakirkham
  - Dante Gama Dessavre (https://github.com/dantegd)

Approvers:
  - Ray Douglass (https://github.com/raydouglass)
  - Dante Gama Dessavre (https://github.com/dantegd)

URL: https://github.com/rapidsai/cuml/pull/6051
---
 ci/test_wheel.sh | 10 +++-------
 1 file changed, 3 insertions(+), 7 deletions(-)

diff --git a/ci/test_wheel.sh b/ci/test_wheel.sh
index 9f473b1890..86eef035cd 100755
--- a/ci/test_wheel.sh
+++ b/ci/test_wheel.sh
@@ -35,13 +35,9 @@ rapids-logger "pytest cuml single GPU"
   -k 'test_sparse_pca_inputs' \
   --junitxml="${RAPIDS_TESTS_DIR}/junit-cuml-sparse-pca.xml"
 
-# Temporarily disabled for CUDA 12.x wheels. 
-# Reference issue: https://github.com/rapidsai/cuml/issues/6050
-if [[ "${RAPIDS_PY_CUDA_SUFFIX}" == "cu11" ]]; then
-  rapids-logger "pytest cuml-dask"
-  ./ci/run_cuml_dask_pytests.sh \
-    --junitxml="${RAPIDS_TESTS_DIR}/junit-cuml-dask.xml"
-fi
+rapids-logger "pytest cuml-dask"
+./ci/run_cuml_dask_pytests.sh \
+  --junitxml="${RAPIDS_TESTS_DIR}/junit-cuml-dask.xml"
 
 rapids-logger "Test script exiting with value: $EXITCODE"
 exit ${EXITCODE}