fix(rust): merge ProjectionOptions for partial cwc

pola-rs · May 21, 2024 · 0be0d6f · 0be0d6f
1 parent 24d8b76
commit 0be0d6f
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/crates/polars-plan/src/logical_plan/optimizer/cluster_with_columns.rs b/crates/polars-plan/src/logical_plan/optimizer/cluster_with_columns.rs
@@ -217,6 +217,10 @@ pub fn optimize(root: Node, lp_arena: &mut Arena<IR>, expr_arena: &Arena<AExpr>)
 
             *current_wc.exprs.exprs_mut() = current_exprs;
 
+            let options = current_wc.options.merge_options(input_wc.options);
+            *current_wc.options = options;
+            *input_wc.options = options;
+
             // @NOTE: Here we add a simple projection to make sure that the output still
             // has the right schema.
             if needs_simple_projection {