From 0be0d6fbd2774937a5a7fc971bed76021f4d7f66 Mon Sep 17 00:00:00 2001
From: coastalwhite <me@gburghoorn.com>
Date: Tue, 21 May 2024 15:23:37 +0200
Subject: [PATCH] fix(rust): merge ProjectionOptions for partial cwc

---
 .../src/logical_plan/optimizer/cluster_with_columns.rs        | 4 ++++
 1 file changed, 4 insertions(+)

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
index 071b6756cb216..8ce6c2d3d8552 100644
--- 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 {