From 2fca5514adbff7148a3be35bbe15cbb8b015c230 Mon Sep 17 00:00:00 2001 From: Ritchie Vink Date: Sat, 23 Mar 2024 14:23:57 +0100 Subject: [PATCH] fix: Ensure ooc_start is set (#15255) --- crates/polars-pipe/src/executors/sinks/sort/sink.rs | 3 +++ 1 file changed, 3 insertions(+) diff --git a/crates/polars-pipe/src/executors/sinks/sort/sink.rs b/crates/polars-pipe/src/executors/sinks/sort/sink.rs index 0f66db59b5a3..1411a76c872d 100644 --- 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::().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