Add timeouts to Dask tests

rapidsai · May 15, 2024 · 89484c3 · 89484c3
1 parent 6be59e8
commit 89484c3
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/ci/run_cuml_dask_pytests.sh b/ci/run_cuml_dask_pytests.sh
@@ -5,10 +5,10 @@
 cd "$(dirname "$(realpath "${BASH_SOURCE[0]}")")"/../python/cuml/tests/dask
 
 rapids-logger "pytest cuml-dask (No UCX-Py/UCXX)"
-python -m pytest --cache-clear "$@" .
+timeout 2h python -m pytest --cache-clear "$@" .
 
 rapids-logger "pytest cuml-dask (UCX-Py only)"
-python -m pytest --cache-clear --run_ucx "$@" .
+timeout 5m python -m pytest --cache-clear --run_ucx "$@" .
 
 rapids-logger "pytest cuml-dask (UCXX only)"
-python -m pytest --cache-clear --run_ucxx "$@" .
+timeout 5m python -m pytest --cache-clear --run_ucxx "$@" .