test whole logisticregressionmg with aligned memory

rapidsai · Mar 13, 2024 · 8ac62fe · 8ac62fe
1 parent da6c99c
commit 8ac62fe
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/ci/run_cuml_dask_pytests.sh b/ci/run_cuml_dask_pytests.sh
@@ -4,4 +4,4 @@
 # Support invoking run_cuml_dask_pytests.sh outside the script directory
 cd "$(dirname "$(realpath "${BASH_SOURCE[0]}")")"/../python/cuml/tests/dask
 
-python -m pytest -s --cache-clear "$@" test_dask_aaa_bug.py
+python -m pytest -s --cache-clear "$@" test_dask_logistic_regression.py
diff --git a/ci/test_python_dask.sh b/ci/test_python_dask.sh
@@ -19,7 +19,7 @@ rapids-logger "pytest cuml-dask"
   --cov-report=xml:"${RAPIDS_COVERAGE_DIR}/cuml-dask-coverage.xml" \
   --cov-report=term \
   -s \
-  test_dask_aaa_bug.py
+  test_dask_logistic_regression.py
 #  .
 
 rapids-logger "Test script exiting with value: $EXITCODE"

diff --git a/ci/test_wheel.sh b/ci/test_wheel.sh
@@ -24,7 +24,7 @@ if [[ "$(arch)" == "aarch64" && "${RAPIDS_BUILD_TYPE}" == "pull-request" ]]; the
     python ci/wheel_smoke_test_cuml.py
 else
     # python -m pytest ./python/cuml/tests -k 'not test_sparse_pca_inputs' -n 4 --ignore=python/cuml/tests/dask && python -m pytest ./python/cuml/tests -k 'test_sparse_pca_inputs' && python -m pytest ./python/cuml/tests/dask
-    python -m pytest -s ./python/cuml/tests/dask/test_dask_aaa_bug.py
+    python -m pytest -s ./python/cuml/tests/dask/test_dask_logistic_regression.py
 fi
 
 rapids-logger "Test script exiting with value: $EXITCODE"