fixup (apache#26657)

mosche · May 11, 2023 · 919e8a4 · 919e8a4
1 parent eb1852a
commit 919e8a4
Showing 1 changed file with 10 additions and 8 deletions.
diff --git a/...ain/java/org/apache/beam/sdk/io/fileschematransform/FileWriteSchemaTransformProvider.java b/...ain/java/org/apache/beam/sdk/io/fileschematransform/FileWriteSchemaTransformProvider.java
@@ -118,14 +118,16 @@ public PCollectionRowTuple expand(PCollectionRowTuple input) {
 
       PCollection<String> files = rowInput.apply("Write Rows", transform);
       PCollection<Row> output =
-          files.apply(
-              "Filenames to Rows",
-              MapElements.into(rows())
-                  .via(
-                      (String name) ->
-                          Row.withSchema(OUTPUT_SCHEMA)
-                              .withFieldValue(FILE_NAME_FIELD.getName(), name)
-                              .build()));
+          files
+              .apply(
+                  "Filenames to Rows",
+                  MapElements.into(rows())
+                      .via(
+                          (String name) ->
+                              Row.withSchema(OUTPUT_SCHEMA)
+                                  .withFieldValue(FILE_NAME_FIELD.getName(), name)
+                                  .build()))
+              .setRowSchema(OUTPUT_SCHEMA);
 
       return PCollectionRowTuple.of(OUTPUT_TAG, output);
     }