FIX correct shape assignment for 3D data (#76)

PSSF23 · web-flow · commit cdcc372d8487 · 2026-04-30T16:58:37.000-04:00
diff --git a/sklearn/tree/_classes.py b/sklearn/tree/_classes.py
@@ -292,7 +292,7 @@ def _fit(
                     )
 
         # Determine output settings
-        n_samples, self.n_features_in_ = X.shape
+        n_samples, self.n_features_in_ = X.shape[0], X.shape[1]
 
         # Do preprocessing if 'y' is passed
         is_classification = False

Original file line number	Diff line number	Diff line change
`@@ -292,7 +292,7 @@ def _fit(`
`292`	`292`	`)`
`293`	`293`
`294`	`294`	`# Determine output settings`
`295`		`- n_samples, self.n_features_in_ = X.shape`
	`295`	`+ n_samples, self.n_features_in_ = X.shape[0], X.shape[1]`
`296`	`296`
`297`	`297`	`# Do preprocessing if 'y' is passed`
`298`	`298`	`is_classification = False`