fix

jonathanc-n · Nov 26, 2024 · 2101112 · 2101112
1 parent ea6f77a
commit 2101112
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/datafusion/physical-plan/src/aggregates/group_values/multi_group_by/primitive.rs b/datafusion/physical-plan/src/aggregates/group_values/multi_group_by/primitive.rs
@@ -191,7 +191,8 @@ impl<T: ArrowPrimitiveType, const NULLABLE: bool> GroupColumn
             assert!(nulls.is_none(), "unexpected nulls in non nullable input");
         }
 
-        let arr = PrimitiveArray::<T>::new(ScalarBuffer::from(group_values), nulls);
+        let arr = PrimitiveArray::<T>::new(ScalarBuffer::from(group_values), nulls)
+            .with_data_type(data_type.clone());
         let array_ref = Arc::new(arr) as ArrayRef;
 
         // Set timezone information for timestamp

diff --git a/datafusion/physical-plan/src/aggregates/group_values/single_group_by/primitive.rs b/datafusion/physical-plan/src/aggregates/group_values/single_group_by/primitive.rs
@@ -209,7 +209,7 @@ where
                 build_primitive(split, null_group)
             }
         };
-        let array_ref = Arc::new(array) as ArrayRef;
+        let array_ref = Arc::new(array.with_data_type(self.data_type.clone())) as ArrayRef;
 
         let adjusted_array = adjust_output_array(&self.data_type, array_ref)
             .expect("Failed to adjust array data type");