support duration return type

shruti2522 · shruti2522 · commit 051e33b08af2 · 2025-04-17T08:21:14.000+05:30
diff --git a/datafusion/expr-common/src/type_coercion/aggregates.rs b/datafusion/expr-common/src/type_coercion/aggregates.rs
@@ -232,6 +232,9 @@ pub fn avg_sum_type(arg_type: &DataType) -> Result<DataType> {
             Ok(DataType::Decimal256(new_precision, *scale))
         }
         arg_type if NUMERICS.contains(arg_type) => Ok(DataType::Float64),
+        DataType::Duration(time_unit) => {
+            Ok(DataType::Duration(*time_unit))
+        }
         DataType::Dictionary(_, dict_value_type) => {
             avg_sum_type(dict_value_type.as_ref())
         }

Original file line number	Diff line number	Diff line change
`@@ -232,6 +232,9 @@ pub fn avg_sum_type(arg_type: &DataType) -> Result<DataType> {`
`232`	`232`	`Ok(DataType::Decimal256(new_precision, *scale))`
`233`	`233`	`}`
`234`	`234`	`arg_type if NUMERICS.contains(arg_type) => Ok(DataType::Float64),`
	`235`	`+ DataType::Duration(time_unit) => {`
	`236`	`+ Ok(DataType::Duration(*time_unit))`
	`237`	`+ }`
`235`	`238`	`DataType::Dictionary(_, dict_value_type) => {`
`236`	`239`	`avg_sum_type(dict_value_type.as_ref())`
`237`	`240`	`}`