Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Jiaming Yuan
2023-12-24 00:09:05 +08:00
committed by GitHub
parent b807f3e30c
commit 6a5f6ba694
4 changed files with 17 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
tests/test_distributed/test_gpu_with_spark/test_data.py
View File

@@ -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

5
tests/test_distributed/test_gpu_with_spark/test_gpu_spark.py
View File

@@ -8,7 +8,10 @@ import sklearn
from xgboost import testing as tm
pytestmark = pytest.mark.skipif(**tm.no_spark())
pytestmark = [
pytest.mark.skipif(**tm.no_spark()),
tm.timeout(240),
]
from pyspark.ml.linalg import Vectors
from pyspark.ml.tuning import CrossValidator, ParamGridBuilder