diff --git a/ci/test_wheel.sh b/ci/test_wheel.sh index e8403559d9..bf3b3845f7 100755 --- a/ci/test_wheel.sh +++ b/ci/test_wheel.sh @@ -22,5 +22,5 @@ python -m pip install $(echo ./dist/cuml*.whl)[test] if [[ "$(arch)" == "aarch64" && "${RAPIDS_BUILD_TYPE}" == "pull-request" ]]; then python ci/wheel_smoke_test_cuml.py else - python -m pytest ./python/cuml/tests -k 'not test_sparse_pca_inputs' -n 8 --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 ./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 fi