Expressions

Signed-off-by: Nicholas Gates <[email protected]>

Author: gatesn

vortex-array/src/expr/expression.rs

@@ -105,6 +105,9 @@ impl Expression {
 
     /// Computes the return dtype of this expression given the input dtype.
     pub fn return_dtype(&self, scope: &DType) -> VortexResult<DType> {
+        if self.is::<Root>() {
+            return Ok(scope.clone());
+        }
         self.scalar_fn.return_dtype(&ExpressionReturnDTypeCtx {
             expr: self,
             scope_dtype: scope,
@@ -247,7 +250,6 @@ impl ReturnDTypeCtx for ExpressionReturnDTypeCtx<'_> {
 
     fn return_dtype(&self, child_idx: usize) -> VortexResult<DType> {
         let child = &self.expr.children()[child_idx];
-
         if child.is::<Root>() {
             return Ok(self.scope_dtype.clone());
         }