fix: Ensure ooc_start is set (pola-rs#15255)

Kylea650 · Mar 23, 2024 · 2fca551 · 2fca551
1 parent bc91c62
commit 2fca551
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/crates/polars-pipe/src/executors/sinks/sort/sink.rs b/crates/polars-pipe/src/executors/sinks/sort/sink.rs
@@ -146,6 +146,9 @@ impl Sink for SortSink {
 
     fn combine(&mut self, other: &mut dyn Sink) {
         let other = other.as_any().downcast_mut::<Self>().unwrap();
+        if let Some(ooc_start) = other.ooc_start {
+            self.ooc_start = Some(ooc_start);
+        }
         self.chunks.extend(std::mem::take(&mut other.chunks));
         self.ooc |= other.ooc;
         self.dist_sample