Fix monotone constraint with tuple input. (#7891)

2022-05-13 04:00:03 +08:00
parent 94ca52b7b7
commit db80671d6b
2 changed files with 8 additions and 1 deletions
--- a/tests/python/test_monotone_constraints.py
+++ b/tests/python/test_monotone_constraints.py
@@ -93,6 +93,11 @@ class TestMonotoneConstraints:
        constrained = xgb.train(params_for_constrained, training_dset)
        assert is_correctly_constrained(constrained)

+    def test_monotone_constraints_tuple(self) -> None:
+        params_for_constrained = {"monotone_constraints": (1, -1)}
+        constrained = xgb.train(params_for_constrained, training_dset)
+        assert is_correctly_constrained(constrained)
+
    @pytest.mark.parametrize('format', [dict, list])
    def test_monotone_constraints_feature_names(self, format):