fix: Throw error for comparison of unequal length series

pola-rs · Sep 18, 2024 · 39f7b1a · 39f7b1a
1 parent 25f84e4
commit 39f7b1a
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/crates/polars-core/src/series/comparison.rs b/crates/polars-core/src/series/comparison.rs
@@ -10,6 +10,8 @@ macro_rules! impl_compare {
         let (lhs, rhs) = ($self, $rhs);
         validate_types(lhs.dtype(), rhs.dtype())?;
 
+        polars_ensure!(lhs.len() == rhs.len(), ShapeMismatch: "could not compare between two series of different length ({} != {})", lhs.len(), rhs.len());
+
         #[cfg(feature = "dtype-categorical")]
         match (lhs.dtype(), rhs.dtype()) {
             (Categorical(_, _) | Enum(_, _), Categorical(_, _) | Enum(_, _)) => {