Add timeouts for CI wheel tests (#1029)

Add timeouts for CI wheel tests to prevent it from getting stuck if something fails in an unexpected way.

Authors:
  - Peter Andreas Entschev (https://github.com/pentschev)

Approvers:
  - Ray Douglass (https://github.com/raydouglass)

URL: #1029

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/