From 26b16d00f8a39e7de77f64ed2a6584d2dc077557 Mon Sep 17 00:00:00 2001
From: Heejong Lee <heejong@gmail.com>
Date: Tue, 10 May 2022 14:05:50 -0700
Subject: [PATCH] [BEAM-14455] Add UUID to sub-schemas for
 PythonExternalTransform

---
 .../beam/sdk/extensions/python/PythonExternalTransform.java     | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/sdks/java/extensions/python/src/main/java/org/apache/beam/sdk/extensions/python/PythonExternalTransform.java b/sdks/java/extensions/python/src/main/java/org/apache/beam/sdk/extensions/python/PythonExternalTransform.java
index c412acd220ee..f9ef62aec577 100644
--- a/sdks/java/extensions/python/src/main/java/org/apache/beam/sdk/extensions/python/PythonExternalTransform.java
+++ b/sdks/java/extensions/python/src/main/java/org/apache/beam/sdk/extensions/python/PythonExternalTransform.java
@@ -170,6 +170,7 @@ Row buildOrGetKwargsRow() {
       Schema schema =
           generateSchemaFromFieldValues(
               kwargsMap.values().toArray(), kwargsMap.keySet().toArray(new String[] {}));
+      schema.setUUID(UUID.randomUUID());
       return Row.withSchema(schema)
           .addValues(convertComplexTypesToRows(kwargsMap.values().toArray()))
           .build();
@@ -223,6 +224,7 @@ private Object[] convertComplexTypesToRows(@Nullable Object @NonNull [] values)
   @VisibleForTesting
   Row buildOrGetArgsRow() {
     Schema schema = generateSchemaFromFieldValues(argsArray, null);
+    schema.setUUID(UUID.randomUUID());
     Object[] convertedValues = convertComplexTypesToRows(argsArray);
     return Row.withSchema(schema).addValues(convertedValues).build();
   }