From 7f2cc5cd7fd38100761dc88f23178fa37cb34c38 Mon Sep 17 00:00:00 2001
From: Francesco Bruzzesi <42817048+FBruzzesi@users.noreply.github.com>
Date: Sun, 19 Jan 2025 20:00:40 +0100
Subject: [PATCH] feat: raise value error for shape mismatch on hierarchical
 (#731)

---
 sklego/meta/hierarchical_predictor.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/sklego/meta/hierarchical_predictor.py b/sklego/meta/hierarchical_predictor.py
index e3835004..0a712dbc 100644
--- a/sklego/meta/hierarchical_predictor.py
+++ b/sklego/meta/hierarchical_predictor.py
@@ -287,6 +287,10 @@ def fit(self, X, y=None):
         global_series = nw.new_series(
             name=self._GLOBAL_NAME, values=np.ones(n_samples), native_namespace=native_namespace
         )
+        if len(target_series) != n_samples:
+            msg = f"Found input variables with inconsistent numbers of samples: {[n_samples, len(target_series)]}"
+            raise ValueError(msg)
+
         frame = X.with_columns(
             **{
                 self._TARGET_NAME: target_series,