make alias condition more restrictive

Eventual-Inc · Oct 4, 2024 · 6d6490b · 6d6490b
1 parent aabd04e
commit 6d6490b
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/src/daft-plan/src/logical_optimization/rules/split_actor_pool_projects.rs b/src/daft-plan/src/logical_optimization/rules/split_actor_pool_projects.rs
@@ -376,7 +376,13 @@ fn try_optimize_project(
     let aliased_projection_exprs = projection
         .projection
         .iter()
-        .map(|e| e.alias(e.name()))
+        .map(|e| {
+            if has_stateful_udf(e) && !matches!(e.as_ref(), Expr::Alias(..)) {
+                e.alias(e.name())
+            } else {
+                e.clone()
+            }
+        })
         .collect();
 
     let aliased_projection = Project::try_new(projection.input.clone(), aliased_projection_exprs)?;