Skip to content

Commit c55d06a

Browse files
authored
Merge pull request #1 from andygrove/ARROW-4749
fix aggr schema
2 parents 9a4988a + f0239c0 commit c55d06a

File tree

1 file changed

+8
-1
lines changed

1 file changed

+8
-1
lines changed

rust/datafusion/src/execution/context.rs

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -185,8 +185,15 @@ impl ExecutionContext {
185185
.collect();
186186
let compiled_aggr_expr = compiled_aggr_expr_result?;
187187

188+
let mut output_fields: Vec<Field> = vec![];
189+
for expr in group_expr {
190+
output_fields.push(expr_to_field(expr, input_schema.as_ref()));
191+
}
192+
for expr in aggr_expr {
193+
output_fields.push(expr_to_field(expr, input_schema.as_ref()));
194+
}
188195
let rel = AggregateRelation::new(
189-
Arc::new(Schema::empty()), //(expr_to_field(&compiled_group_expr, &input_schema))),
196+
Arc::new(Schema::new(output_fields)),
190197
input_rel,
191198
compiled_group_expr,
192199
compiled_aggr_expr,

0 commit comments

Comments
 (0)