From e3eb80fa515200377328bc18e39ff9a083f43e59 Mon Sep 17 00:00:00 2001 From: Andrew Lamb Date: Mon, 5 Aug 2024 18:06:03 -0400 Subject: [PATCH] workaround sum bug --- .../src/aggregate/groups_accumulator/prim_op.rs | 2 ++ 1 file changed, 2 insertions(+) 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 index 8d69646bd422a..1fa9cef94d32a 100644 --- 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)]) }