Fix python api

adam2392 · adam2392 · commit 90b4a552c211 · 2023-02-02T23:40:19.000-05:00
Signed-off-by: Adam Li &lt;adam2392@gmail.com&gt;
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()
 
