From 8c5e2164fa1bd010a59904a84c4e04cc79653d19 Mon Sep 17 00:00:00 2001
From: Rory Mitchell <r.a.mitchell.nz@gmail.com>
Date: Wed, 13 Nov 2024 03:44:10 -0800
Subject: [PATCH] Add n_estimators

---
 legateboost/legateboost.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/legateboost/legateboost.py b/legateboost/legateboost.py
index a56af6ae..80b3edc8 100644
--- 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,