Skip to content

Commit 24fbc63

Browse files
andygrovenevi-me
authored andcommitted
fix aggr schema
1 parent 217dafb commit 24fbc63

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
@@ -187,8 +187,15 @@ impl ExecutionContext {
187187
.collect();
188188
let compiled_aggr_expr = compiled_aggr_expr_result?;
189189

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

0 commit comments

Comments
 (0)