Fix python api

Signed-off-by: Adam Li <[email protected]>
neurodata · Feb 3, 2023 · 90b4a55 · 90b4a55
1 parent c30f54d
commit 90b4a55
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/sklearn/tree/_classes.py b/sklearn/tree/_classes.py
@@ -190,9 +190,10 @@ def fit(self, X, y=None, sample_weight=None, check_input=True):
             # csr.
             check_X_params = dict(dtype=DTYPE, accept_sparse="csc")
             check_y_params = dict(ensure_2d=False, dtype=None)
-            X = self._validate_data(X, validate_separately=check_X_params)
             if y is not None:
-                y = self._validate_data(y, validate_separately=check_y_params)
+                X, y = self._validate_data(X, y, validate_separately=(check_X_params, check_y_params))
+            else:
+                X = self._validate_data(X, **check_X_params)
             if issparse(X):
                 X.sort_indices()