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

Mitigate flaky GPU test. (#8078)

Browse Source 
The flakiness is caused by the global random engine, which will take some time to fix.
This commit is contained in:
Jiaming Yuan 2022-07-16 13:45:32 +08:00 committed by GitHub
parent 7a5586f3db
commit 0ce80b7bcf
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/python-gpu/test_gpu_updaters.py
View File

@ -114,7 +114,7 @@ class TestGPUUpdaters:
param = dataset.set_params(param) param = dataset.set_params(param)
result = train_result(param, dataset.get_device_dmat(), num_rounds) result = train_result(param, dataset.get_device_dmat(), num_rounds)
note(result) note(result)
assert tm.non_increasing(result['train'][dataset.metric]) assert tm.non_increasing(result['train'][dataset.metric], tolerance=1e-3)
@given(parameter_strategy, strategies.integers(1, 20), @given(parameter_strategy, strategies.integers(1, 20),
tm.dataset_strategy) tm.dataset_strategy)