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

koaning · Jan 17, 2025 · 9c53551 · 9c53551
1 parent 9b15cef
commit 9c53551
Showing 1 changed file with 4 additions and 0 deletions.
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)