[CI] Add timeout for distributed GPU tests. (#9917)

2023-12-24 00:09:05 +08:00
parent b807f3e30c
commit 6a5f6ba694
4 changed files with 17 additions and 7 deletions
--- a/tests/test_distributed/test_gpu_with_spark/test_data.py
+++ b/tests/test_distributed/test_gpu_with_spark/test_data.py
@@ -2,7 +2,10 @@ import pytest

 from xgboost import testing as tm

-pytestmark = pytest.mark.skipif(**tm.no_spark())
+pytestmark = [
+    pytest.mark.skipif(**tm.no_spark()),
+    tm.timeout(120),
+]

 from ..test_with_spark.test_data import run_dmatrix_ctor