[CHORE] FunctionExpr and exp (#2119)

Eventual-Inc · Apr 15, 2024 · b4a6680 · b4a6680
1 parent 69a1f5c
commit b4a6680
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/daft-dsl/src/functions/numeric/exp.rs b/src/daft-dsl/src/functions/numeric/exp.rs
@@ -3,7 +3,7 @@ use daft_core::datatypes::Field;
 use daft_core::schema::Schema;
 use daft_core::{DataType, Series};
 
-use crate::functions::FunctionEvaluator;
+use crate::functions::{FunctionEvaluator, FunctionExpr};
 use crate::Expr;
 
 pub(super) struct ExpEvaluator {}
@@ -13,7 +13,7 @@ impl FunctionEvaluator for ExpEvaluator {
         "exp"
     }
 
-    fn to_field(&self, inputs: &[Expr], schema: &Schema, _: &Expr) -> DaftResult<Field> {
+    fn to_field(&self, inputs: &[Expr], schema: &Schema, _: &FunctionExpr) -> DaftResult<Field> {
         if inputs.len() != 1 {
             return Err(DaftError::SchemaMismatch(format!(
                 "Expected 1 input arg, got {}",
@@ -34,7 +34,7 @@ impl FunctionEvaluator for ExpEvaluator {
         Ok(Field::new(field.name, dtype))
     }
 
-    fn evaluate(&self, inputs: &[Series], _: &Expr) -> DaftResult<Series> {
+    fn evaluate(&self, inputs: &[Series], _: &FunctionExpr) -> DaftResult<Series> {
         if inputs.len() != 1 {
             return Err(DaftError::SchemaMismatch(format!(
                 "Expected 1 input arg, got {}",