bump sklearn to 1.6 and rename deprecated parameters (#377)

Fix #376

Author: mariushelf

ngboost/ngboost.py

@@ -381,7 +381,7 @@ def partial_fit(
             X,
             Y,
             accept_sparse=True,
-            force_all_finite="allow-nan",
+            ensure_all_finite="allow-nan",
             multi_output=self.multi_output,
             y_numeric=True,
         )
@@ -398,7 +398,7 @@ def partial_fit(
                 X_val,
                 Y_val,
                 accept_sparse=True,
-                force_all_finite="allow-nan",
+                ensure_all_finite="allow-nan",
                 multi_output=self.multi_output,
                 y_numeric=True,
             )
@@ -532,7 +532,7 @@ def pred_dist(self, X, max_iter=None):
             A NGBoost distribution object
         """
 
-        X = check_array(X, accept_sparse=True, force_all_finite="allow-nan")
+        X = check_array(X, accept_sparse=True, ensure_all_finite="allow-nan")
 
         params = np.asarray(self.pred_param(X, max_iter))
         dist = self.Dist(params.T)
@@ -579,7 +579,7 @@ def predict(self, X, max_iter=None):
             Numpy array of the estimates of Y
         """
 
-        X = check_array(X, accept_sparse=True, force_all_finite="allow-nan")
+        X = check_array(X, accept_sparse=True, ensure_all_finite="allow-nan")
 
         return self.pred_dist(X, max_iter=max_iter).predict()
 

pyproject.toml

@@ -14,7 +14,7 @@ license = "Apache License 2.0"
 
 [tool.poetry.dependencies]
 python = ">=3.9, <3.13"
-scikit-learn = ">=1.0.2"
+scikit-learn = "^1.6"
 numpy = ">=1.21.2"
 scipy = ">=1.7.2"
 tqdm = ">=4.3"