Fix sklearn tests (#280)

ray-project · Apr 24, 2023 · 02a1cb3 · 02a1cb3
1 parent 648d6dd
commit 02a1cb3
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/xgboost_ray/tests/test_sklearn.py b/xgboost_ray/tests/test_sklearn.py
@@ -976,10 +976,8 @@ def test_constraint_parameters(self):
         reg.fit(X, y)
 
         config = json.loads(reg.get_booster().save_config())
-        if XGBOOST_VERSION > Version("1.7.4"):
-            assert (config["learner"]["gradient_booster"]["tree_train_param"][
-                "interaction_constraints"] == "[[0, 1], [2, 3, 4]]")
-        elif XGBOOST_VERSION >= Version("1.6.0"):
+
+        if XGBOOST_VERSION >= Version("1.6.0"):
             assert (config["learner"]["gradient_booster"]["updater"][
                 "grow_histmaker"]["train_param"]["interaction_constraints"] ==
                     "[[0, 1], [2, 3, 4]]")