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

Time the CPU tests on Jenkins. (#6257)

Browse Source 
* Time the CPU tests on Jenkins.
* Reduce thread contention.
* Add doc.
* Skip heavy tests on ARM.
This commit is contained in:
Jiaming Yuan
2020-10-21 08:19:07 +08:00
committed by GitHub
parent d1254808d5
commit 81c37c28d5
9 changed files with 29 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
tests/python/test_with_dask.py
View File

@@ -611,10 +611,6 @@ class TestWithDask:
tree_method):
params['tree_method'] = tree_method
params = dataset.set_params(params)
# multi class doesn't handle empty dataset well (empty
# means at least 1 worker has data).
if params['objective'] == "multi:softmax":
return
# It doesn't make sense to distribute a completely
# empty dataset.
if dataset.X.shape[0] == 0:
@@ -640,18 +636,20 @@ class TestWithDask:
# Make sure that it's decreasing
assert history[-1] < history[0]
@pytest.mark.skipif(**tm.is_arm())
@given(params=hist_parameter_strategy,
num_rounds=strategies.integers(20, 30),
dataset=tm.dataset_strategy)
@settings(deadline=None)
def test_hist(self, params, num_rounds, dataset, client):
def test_hist(self, params, dataset, client):
num_rounds = 30
self.run_updater_test(client, params, num_rounds, dataset, 'hist')
@pytest.mark.skipif(**tm.is_arm())
@given(params=exact_parameter_strategy,
num_rounds=strategies.integers(20, 30),
dataset=tm.dataset_strategy)
@settings(deadline=None)
def test_approx(self, client, params, num_rounds, dataset):
def test_approx(self, client, params, dataset):
num_rounds = 30
self.run_updater_test(client, params, num_rounds, dataset, 'approx')
def run_quantile(self, name):