fix bug

apache · Nov 19, 2019 · 815529d · 815529d
1 parent 48bf173
commit 815529d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sdks/java/core/src/main/java/org/apache/beam/sdk/schemas/transforms/SchemaAggregateFn.java b/sdks/java/core/src/main/java/org/apache/beam/sdk/schemas/transforms/SchemaAggregateFn.java
@@ -152,7 +152,7 @@ Inner withSchema(Schema inputSchema) {
  FieldAggregation fieldAggregation = fieldAggregations.get(i);
  SimpleFunction<Row, ?> extractFunction;
  Coder extractOutputCoder;
- if (fieldAggregation.unnestedInputSubSchema.getFieldCount() == 1) {
+ if (fieldAggregation.fieldsToAggregate.referencesSingleField()) {
  extractFunction = new ExtractSingleFieldFunction(fieldAggregation);
  extractOutputCoder =
  SchemaCoder.coderForFieldType(