refactor: replace ensure_expr with ensure_expr_list for aggregation expressions

kosiew · kosiew · commit 4dfcaed4f46a · 2025-09-02T13:59:07.000+08:00
diff --git a/python/datafusion/dataframe.py b/python/datafusion/dataframe.py
@@ -539,7 +539,7 @@ def aggregate(
         )
 
         group_by_exprs = expr_list_to_raw_expr_list(group_by_list)
-        aggs_exprs = [ensure_expr(agg) for agg in aggs_list]
+        aggs_exprs = ensure_expr_list(aggs_list)
         return DataFrame(self.df.aggregate(group_by_exprs, aggs_exprs))
 
     def sort(self, *exprs: SortKey) -> DataFrame:

Original file line number	Diff line number	Diff line change
`@@ -539,7 +539,7 @@ def aggregate(`
`539`	`539`	`)`
`540`	`540`
`541`	`541`	`group_by_exprs = expr_list_to_raw_expr_list(group_by_list)`
`542`		`- aggs_exprs = [ensure_expr(agg) for agg in aggs_list]`
	`542`	`+ aggs_exprs = ensure_expr_list(aggs_list)`
`543`	`543`	`return DataFrame(self.df.aggregate(group_by_exprs, aggs_exprs))`
`544`	`544`
`545`	`545`	`def sort(self, *exprs: SortKey) -> DataFrame:`