workaround sum bug

alamb · alamb · commit e3eb80fa5152 · 2024-08-05T18:06:03.000-04:00
diff --git a/datafusion/physical-expr-common/src/aggregate/groups_accumulator/prim_op.rs b/datafusion/physical-expr-common/src/aggregate/groups_accumulator/prim_op.rs
@@ -185,6 +185,8 @@ where
                 )
             })?
             .map_err(DataFusionError::from)?;
+        let state_values = state_values
+            .with_data_type(self.data_type.clone());
 
         Ok(vec![Arc::new(state_values)])
     }

Original file line number	Diff line number	Diff line change
`@@ -185,6 +185,8 @@ where`
`185`	`185`	`)`
`186`	`186`	`})?`
`187`	`187`	`.map_err(DataFusionError::from)?;`
	`188`	`+ let state_values = state_values`
	`189`	`+ .with_data_type(self.data_type.clone());`
`188`	`190`
`189`	`191`	`Ok(vec![Arc::new(state_values)])`
`190`	`192`	`}`