From 89484c353adc6b3b8faf854726217bc042c1cd1d Mon Sep 17 00:00:00 2001 From: Peter Andreas Entschev Date: Tue, 14 May 2024 05:29:59 -0700 Subject: [PATCH] Add timeouts to Dask tests --- ci/run_cuml_dask_pytests.sh | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/ci/run_cuml_dask_pytests.sh b/ci/run_cuml_dask_pytests.sh index 93ace5922e..0472c147fa 100755 --- 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 "$@" .