propagate targetParallelism from SortedBucketIO (#2997)

😅
spotify · May 21, 2020 · 2c1b71b · 2c1b71b
1 parent f853138
commit 2c1b71b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scio-smb/src/main/java/org/apache/beam/sdk/extensions/smb/SortedBucketIO.java b/scio-smb/src/main/java/org/apache/beam/sdk/extensions/smb/SortedBucketIO.java
@@ -105,7 +105,7 @@ public <V> CoGbkTransform<K, V> to(SortedBucketIO.Write<K, V> write) {
     public PCollection<KV<K, CoGbkResult>> expand(PBegin input) {
       List<BucketedInput<?, ?>> bucketedInputs =
           reads.stream().map(Read::toBucketedInput).collect(Collectors.toList());
-      return input.apply(new SortedBucketSource<>(keyClass, bucketedInputs));
+      return input.apply(new SortedBucketSource<>(keyClass, bucketedInputs, targetParallelism));
     }
   }