Reduce regularization in GPU gblinear test. (#8010)

dmlc · Jun 20, 2022 · d285d6b · d285d6b
1 parent e58e417
commit d285d6b
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/tests/python-gpu/test_gpu_linear.py b/tests/python-gpu/test_gpu_linear.py
@@ -42,9 +42,13 @@ def test_gpu_coordinate(self, param, num_rounds, dataset):
     # Loss is not guaranteed to always decrease because of regularisation parameters
     # We test a weaker condition that the loss has not increased between the first and last
     # iteration
-    @given(parameter_strategy, strategies.integers(10, 50),
-           tm.dataset_strategy, strategies.floats(1e-5, 1.0),
-           strategies.floats(1e-5, 1.0))
+    @given(
+        parameter_strategy,
+        strategies.integers(10, 50),
+        tm.dataset_strategy,
+        strategies.floats(1e-5, 0.8),
+        strategies.floats(1e-5, 0.8)
+    )
     @settings(deadline=None, print_blob=True)
     def test_gpu_coordinate_regularised(self, param, num_rounds, dataset, alpha, lambd):
         assume(len(dataset.y) > 0)