Fix monotone constraint with tuple input. (#7891) (#8159)

2022-08-12 22:05:53 +08:00
parent 140c377a96
commit 9c653378e2
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):