feat: raise value error for shape mismatch (#730)

FBruzzesi · web-flow · commit 9c53551464d4 · 2025-01-17T15:05:23.000+01:00
diff --git a/sklego/meta/_grouped_utils.py b/sklego/meta/_grouped_utils.py
@@ -37,6 +37,10 @@ def parse_X_y(X, y, groups, check_X=True, **kwargs) -> nw.DataFrame:
         name="tmp", values=[None] * n_samples if y is None else y, native_namespace=nw.get_native_namespace(X)
     )
 
+    if len(y_series) != n_samples:
+        msg = f"Found input variables with inconsistent numbers of samples: {[n_samples, len(y_series)]}"
+        raise ValueError(msg)
+
     return X.with_columns(__sklego_target__=y_series)
 
 

Original file line number	Diff line number	Diff line change
`@@ -37,6 +37,10 @@ def parse_X_y(X, y, groups, check_X=True, **kwargs) -> nw.DataFrame:`
`37`	`37`	`name="tmp", values=[None] * n_samples if y is None else y, native_namespace=nw.get_native_namespace(X)`
`38`	`38`	`)`
`39`	`39`
	`40`	`+ if len(y_series) != n_samples:`
	`41`	`+ msg = f"Found input variables with inconsistent numbers of samples: {[n_samples, len(y_series)]}"`
	`42`	`+ raise ValueError(msg)`
	`43`	`+`
`40`	`44`	`return X.with_columns(__sklego_target__=y_series)`
`41`	`45`
`42`	`46`