Add n_estimators

rapidsai · Nov 13, 2024 · 8c5e216 · 8c5e216
1 parent 79735e3
commit 8c5e216
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/legateboost/legateboost.py b/legateboost/legateboost.py
@@ -61,7 +61,7 @@ def __init__(
                 "model_init_": AddMember.ADD,
                 "n_features_in_": AddMember.ASSERT_SAME,
                 "is_fitted_": AddMember.ASSERT_SAME,
-                "n_estimators": AddMember.PREFER_A,
+                "n_estimators": AddMember.ADD,
                 "objective": AddMember.ASSERT_SAME,
                 "metric": AddMember.PREFER_A,
                 "learning_rate": AddMember.PREFER_A,