[BEAM-14455] Add UUID to sub-schemas for PythonExternalTransform

tomstepp · May 10, 2022 · 26b16d0 · 26b16d0
1 parent 1215973
commit 26b16d0
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
@@ -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();
   }