Skip to content
This repository was archived by the owner on Jan 12, 2026. It is now read-only.

Commit 02a1cb3

Browse files
authored
Fix sklearn tests (#280)
1 parent 648d6dd commit 02a1cb3

File tree

1 file changed

+2
-4
lines changed

1 file changed

+2
-4
lines changed

xgboost_ray/tests/test_sklearn.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -976,10 +976,8 @@ def test_constraint_parameters(self):
976976
reg.fit(X, y)
977977

978978
config = json.loads(reg.get_booster().save_config())
979-
if XGBOOST_VERSION > Version("1.7.4"):
980-
assert (config["learner"]["gradient_booster"]["tree_train_param"][
981-
"interaction_constraints"] == "[[0, 1], [2, 3, 4]]")
982-
elif XGBOOST_VERSION >= Version("1.6.0"):
979+
980+
if XGBOOST_VERSION >= Version("1.6.0"):
983981
assert (config["learner"]["gradient_booster"]["updater"][
984982
"grow_histmaker"]["train_param"]["interaction_constraints"] ==
985983
"[[0, 1], [2, 3, 4]]")

0 commit comments

Comments
 (0)