Return PhysicalGroupBy from AggregateExec::group_expr

apache · Jun 13, 2022 · a2cb52d · a2cb52d
1 parent 27dcdb1
commit a2cb52d
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/datafusion/core/src/physical_plan/aggregates/mod.rs b/datafusion/core/src/physical_plan/aggregates/mod.rs
@@ -203,8 +203,8 @@ impl AggregateExec {
     }
 
     /// Grouping expressions
-    pub fn group_expr(&self) -> &[(Arc<dyn PhysicalExpr>, String)] {
-        self.group_by.expr()
+    pub fn group_expr(&self) -> &PhysicalGroupBy {
+        &self.group_by
     }
 
     /// Grouping expressions as they occur in the output schema
@@ -757,7 +757,7 @@ mod tests {
         ];
         assert_batches_sorted_eq!(expected, &result);
 
-        let groups = partial_aggregate.group_expr().to_vec();
+        let groups = partial_aggregate.group_expr().expr().to_vec();
 
         let merge = Arc::new(CoalescePartitionsExec::new(partial_aggregate));