Merge pull request apache#17605 from ihji/BEAM-14455

[BEAM-14455] Add UUID to sub-schemas for PythonExternalTransform
tomstepp · May 13, 2022 · 04f4984 · 04f4984
2 parents 0a2aed7 + 26b16d0
commit 04f4984
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/...s/python/src/main/java/org/apache/beam/sdk/extensions/python/PythonExternalTransform.java b/...s/python/src/main/java/org/apache/beam/sdk/extensions/python/PythonExternalTransform.java
@@ -199,6 +199,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();
@@ -254,6 +255,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();
   }