From a0cf3637cd712ecefd3b58a1ae53e1638fb35d3a Mon Sep 17 00:00:00 2001
From: Danny Mccormick <dannymccormick@google.com>
Date: Wed, 18 Dec 2024 16:57:21 -0500
Subject: [PATCH] Fix up

---
 .../beam/sdk/io/aws2/kinesis/KinesisTransformRegistrar.java  | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/sdks/java/io/amazon-web-services2/src/main/java/org/apache/beam/sdk/io/aws2/kinesis/KinesisTransformRegistrar.java b/sdks/java/io/amazon-web-services2/src/main/java/org/apache/beam/sdk/io/aws2/kinesis/KinesisTransformRegistrar.java
index 8bbe9968024c..8df97f6d5304 100644
--- a/sdks/java/io/amazon-web-services2/src/main/java/org/apache/beam/sdk/io/aws2/kinesis/KinesisTransformRegistrar.java
+++ b/sdks/java/io/amazon-web-services2/src/main/java/org/apache/beam/sdk/io/aws2/kinesis/KinesisTransformRegistrar.java
@@ -32,7 +32,6 @@
 import org.apache.beam.sdk.transforms.PTransform;
 import org.apache.beam.sdk.values.PBegin;
 import org.apache.beam.sdk.values.PCollection;
-import org.apache.beam.sdk.values.PDone;
 import org.apache.beam.vendor.guava.v32_1_2_jre.com.google.common.collect.ImmutableMap;
 import org.checkerframework.checker.nullness.qual.Nullable;
 import org.joda.time.Duration;
@@ -86,7 +85,7 @@ public void setServiceEndpoint(@Nullable String serviceEndpoint) {
   }
 
   public static class WriteBuilder
-      implements ExternalTransformBuilder<WriteBuilder.Configuration, PCollection<byte[]>, PDone> {
+      implements ExternalTransformBuilder<WriteBuilder.Configuration, PCollection<byte[]>, KinesisIO.Write.Result> {
 
     public static class Configuration extends CrossLanguageConfiguration {
       private String partitionKey;
@@ -97,7 +96,7 @@ public void setPartitionKey(String partitionKey) {
     }
 
     @Override
-    public PTransform<PCollection<byte[]>, PDone> buildExternal(Configuration configuration) {
+    public PTransform<PCollection<byte[]>, KinesisIO.Write.Result> buildExternal(Configuration configuration) {
       AwsBasicCredentials creds = AwsBasicCredentials.create(configuration.awsAccessKey, configuration.awsSecretKey);
       StaticCredentialsProvider provider = StaticCredentialsProvider.create(creds);
       KinesisIO.Write<byte[]> writeTransform =