wip

Author: jayshrivastava

datafusion/physical-plan/src/aggregates/mod.rs

@@ -1473,11 +1473,9 @@ impl ExecutionPlan for AggregateExec {
         // This optimization is NOT safe for filters on aggregated columns (like filtering on
         // the result of SUM or COUNT), as those require computing all groups first.
 
-        // Build grouping columns using OUTPUT indices, not input indices.
-        // Parent filters reference the AggregateExec's output schema where grouping
-        // columns occupy positions [0..num_groups). The grouping expressions reference
-        // INPUT columns which may have different indices (e.g., when an intermediate
-        // ProjectionExec reorders columns). We must compare in the same index space.
+        // Build grouping columns using output indices because parent filters reference the AggregateExec's output schema where grouping
+        // columns in the output schema. The grouping expressions reference
+        // input columns which may not match the output schema.
         let output_schema = self.schema();
         let grouping_columns: HashSet<_> = (0..self.group_by.expr().len())
             .map(|i| Column::new(output_schema.field(i).name(), i))