FIX update impute_missing default

neurodata · Jul 9, 2024 · 06ba512 · 06ba512
1 parent 19ed6f3
commit 06ba512
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/treeple/ensemble/_honest_forest.py b/treeple/ensemble/_honest_forest.py
@@ -648,7 +648,7 @@ def predict_proba(self, X):
         """
         return self._predict_proba(X)
 
-    def _predict_proba(self, X, indices=None, impute_missing=None):
+    def _predict_proba(self, X, indices=None, impute_missing=np.nan):
         """predict_proba helper class"""
         check_is_fitted(self)
         X = self._validate_X_predict(X)