(bugfix) Set metadata in AvroSortedBucketIO (#5184)

spotify · Jan 18, 2024 · bc4ecfb · bc4ecfb
1 parent c2886ee
commit bc4ecfb
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/scio-smb/src/main/java/org/apache/beam/sdk/extensions/smb/AvroSortedBucketIO.java b/scio-smb/src/main/java/org/apache/beam/sdk/extensions/smb/AvroSortedBucketIO.java
@@ -382,7 +382,9 @@ public Write<K1, K2, T> withTempDirectory(String tempDirectory) {
     @SuppressWarnings("unchecked")
     @Override
     public FileOperations<T> getFileOperations() {
-      return AvroFileOperations.of(getDatumFactory(), getSchema()).withCodec(getCodec());
+      return AvroFileOperations.of(getDatumFactory(), getSchema())
+          .withCodec(getCodec())
+          .withMetadata(getMetadata());
     }
 
     @SuppressWarnings("unchecked")