Mitigate flaky GPU test. (#8078)

The flakiness is caused by the global random engine, which will take some time to fix.
2022-07-16 13:45:32 +08:00 · 2022-07-16 13:45:32 +08:00 · 0ce80b7bcf
commit 0ce80b7bcf
parent 7a5586f3db
1 changed files with 1 additions and 1 deletions
--- a/tests/python-gpu/test_gpu_updaters.py
+++ b/tests/python-gpu/test_gpu_updaters.py
@ -114,7 +114,7 @@ class TestGPUUpdaters:
        param = dataset.set_params(param)
        result = train_result(param, dataset.get_device_dmat(), num_rounds)
        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),
           tm.dataset_strategy)