Add timeouts for CI wheel tests

rapidsai · Mar 4, 2024 · ad3725f · ad3725f
1 parent 388c7af
commit ad3725f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/ci/test_wheel.sh b/ci/test_wheel.sh
@@ -11,6 +11,6 @@ RAPIDS_PY_WHEEL_NAME="ucx_py_${RAPIDS_PY_CUDA_SUFFIX}" rapids-download-wheels-fr
 python -m pip install $(echo ./dist/ucx_py*.whl)[test]
 
 cd tests
-python -m pytest --cache-clear -vs .
+timeout 10m python -m pytest --cache-clear -vs .
 cd ../ucp
-python -m pytest --cache-clear -vs ./_libs/tests/
+timeout 2m python -m pytest --cache-clear -vs ./_libs/tests/