diff --git a/xgboost_ray/tests/test_sklearn.py b/xgboost_ray/tests/test_sklearn.py index 84a70e3f..c87390c5 100644 --- 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]]")