From bc4ecfb2e326fa279fff8b4a519db2710085a318 Mon Sep 17 00:00:00 2001
From: Claire McGinty <clairem@spotify.com>
Date: Thu, 18 Jan 2024 13:02:46 -0500
Subject: [PATCH] (bugfix) Set metadata in AvroSortedBucketIO (#5184)

---
 .../apache/beam/sdk/extensions/smb/AvroSortedBucketIO.java    | 4 +++-
 1 file changed, 3 insertions(+), 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
index a49531b4f4..d084ae30ed 100644
--- 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")