hypothesis error.

trivialfis · Jun 29, 2023 · 0190537 · 0190537
1 parent 19dc937
commit 0190537
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/python-package/xgboost/testing/params.py b/python-package/xgboost/testing/params.py
@@ -5,7 +5,7 @@
 import pytest
 
 strategies = pytest.importorskip("hypothesis.strategies")
-
+max_leaves = [2**i + 1 for i in range(0, 10)] + [0]
 
 exact_parameter_strategy = strategies.fixed_dictionaries(
     {
@@ -27,7 +27,7 @@
 hist_parameter_strategy = strategies.fixed_dictionaries(
     {
         "max_depth": strategies.integers(1, 11),
-        "max_leaves": strategies.integers(0, 1024),
+        "max_leaves": strategies.sampled_from(max_leaves),
         "max_bin": strategies.integers(2, 512),
         "grow_policy": strategies.sampled_from(["lossguide", "depthwise"]),
         "min_child_weight": strategies.floats(0.5, 2.0),
@@ -44,7 +44,7 @@
 hist_multi_parameter_strategy = strategies.fixed_dictionaries(
     {
         "max_depth": strategies.integers(1, 11),
-        "max_leaves": strategies.integers(0, 1024),
+        "max_leaves": strategies.sampled_from(max_leaves),
         "max_bin": strategies.integers(2, 512),
         "multi_strategy": strategies.sampled_from(
             ["multi_output_tree", "one_output_per_tree"]