fix Expr::name for struct get

Eventual-Inc · Feb 2, 2024 · 58bb051 · 58bb051
1 parent 59af587
commit 58bb051
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/src/daft-dsl/src/expr.rs b/src/daft-dsl/src/expr.rs
@@ -7,7 +7,7 @@ use daft_core::{
 };
 
 use crate::{
-    functions::FunctionEvaluator,
+    functions::{struct_::StructExpr, FunctionEvaluator},
     lit,
     optimization::{get_required_columns, requires_computation},
 };
@@ -524,7 +524,10 @@ impl Expr {
             NotNull(expr) => expr.name(),
             IsIn(expr, ..) => expr.name(),
             Literal(..) => Ok("literal"),
-            Function { func: _, inputs } => inputs.first().unwrap().name(),
+            Function { func, inputs } => match func {
+                FunctionExpr::Struct(StructExpr::Get(name)) => Ok(name),
+                _ => inputs.first().unwrap().name(),
+            },
             BinaryOp {
                 op: _,
                 left,