fix: Fix object chunked gather (#19811)

pola-rs · Nov 15, 2024 · 9ceedde · 9ceedde
1 parent 989eade
commit 9ceedde
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/crates/polars-core/src/series/implementations/object.rs b/crates/polars-core/src/series/implementations/object.rs
@@ -132,11 +132,13 @@ where
     }
 
     fn take(&self, indices: &IdxCa) -> PolarsResult<Series> {
-        Ok(self.0.take(indices)?.into_series())
+        let ca = self.rechunk_object();
+        Ok(ca.take(indices)?.into_series())
     }
 
     unsafe fn take_unchecked(&self, indices: &IdxCa) -> Series {
-        self.0.take_unchecked(indices).into_series()
+        let ca = self.rechunk_object();
+        ca.take_unchecked(indices).into_series()
     }
 
     fn take_slice(&self, indices: &[IdxSize]) -> PolarsResult<Series> {