Break from the outer loop.

pola-rs · Oct 21, 2023 · 2537769 · 2537769
1 parent dcda8a4
commit 2537769
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/crates/polars-pipe/src/pipeline/dispatcher.rs b/crates/polars-pipe/src/pipeline/dispatcher.rs
@@ -537,7 +537,7 @@ fn run_pipeline_no_finalize_iterative(
             operator_end: usize,
             sink_finished: &mut bool,
         ) -> PolarsResult<(u32, Box<dyn Sink>)> {
-            for src in &mut std::mem::take(&mut pipeline.sources) {
+            'outer: for src in &mut std::mem::take(&mut pipeline.sources) {
                 let mut next_batches = src.get_batches(ec)?;
 
                 while let SourceResult::GotMoreData(chunks) = next_batches {
@@ -553,7 +553,7 @@ fn run_pipeline_no_finalize_iterative(
 
                     if let Some(SinkResult::Finished) = sink_result {
                         *sink_finished = true;
-                        break;
+                        break 'outer;
                     }
                 }
             }