chore(cubestore): Use fourth input parameter for Sum::new

rust/cubestore/cubestore/src/metastore/table.rs

@@ -8,10 +8,10 @@
 use crate::rocks_table_impl;
 use crate::{base_rocks_secondary_index, CubeError};
 use byteorder::{BigEndian, WriteBytesExt};
 use chrono::DateTime;
 use chrono::Utc;
 use datafusion::arrow::datatypes::Schema as ArrowSchema;
-use datafusion::physical_plan::expressions::{Column as FusionColumn, Max, Min, Sum};
+use datafusion::physical_plan::expressions::{sum_return_type, Column as FusionColumn, Max, Min, Sum};
 use datafusion::physical_plan::{udaf, AggregateExpr, PhysicalExpr};
 use itertools::Itertools;
 
@@ -74,9 +74,10 @@
             self.column.get_name().as_str(),
             &schema,
         )?);
         let res: Arc<dyn AggregateExpr> = match self.function {
             AggregateFunction::SUM => {
-                Arc::new(Sum::new(col.clone(), col.name(), col.data_type(schema)?))
+                let input_data_type = col.data_type(schema)?;
+                Arc::new(Sum::new(col.clone(), col.name(), sum_return_type(&input_data_type)?, &input_data_type))
             }
             AggregateFunction::MAX => {
                 Arc::new(Max::new(col.clone(), col.name(), col.data_type(schema)?))