From f7f4f6278fa50b4953f9bd2b3a32b85bb76a42aa Mon Sep 17 00:00:00 2001
From: weioren <65381401+weioren@users.noreply.github.com>
Date: Mon, 21 Apr 2025 03:42:40 -0400
Subject: [PATCH] allow nans in _forest.py

---
 sklearn/ensemble/_forest.py | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

diff --git a/sklearn/ensemble/_forest.py b/sklearn/ensemble/_forest.py
index 99aa86157d6e9..5972e46d76346 100644
--- a/sklearn/ensemble/_forest.py
+++ b/sklearn/ensemble/_forest.py
@@ -774,11 +774,13 @@ def _validate_X_predict(self, X):
         """
         Validate X whenever one tries to predict, apply, predict_proba."""
         check_is_fitted(self)
-        if self.estimators_[0]._support_missing_values(X):
-            ensure_all_finite = "allow-nan"
-        else:
-            ensure_all_finite = True
+        # if self.estimators_[0]._support_missing_values(X):
+        #     ensure_all_finite = "allow-nan"
+        # else:
+        #     ensure_all_finite = True
 
+        # Always allow NaNs:
+        ensure_all_finite = "allow-nan"
         X = validate_data(
             self,
             X,