diff --git a/src/main/java/com/google/cloud/pubsublite/spark/PslSparkUtils.java b/src/main/java/com/google/cloud/pubsublite/spark/PslSparkUtils.java index 066a1eae..cbf7969b 100644 --- a/src/main/java/com/google/cloud/pubsublite/spark/PslSparkUtils.java +++ b/src/main/java/com/google/cloud/pubsublite/spark/PslSparkUtils.java @@ -176,7 +176,7 @@ public static void verifyWriteInputSchema(StructType inputSchema) { Option idxOr = inputSchema.getFieldIndex(k); if (!idxOr.isEmpty()) { StructField f = inputSchema.apply((int) idxOr.get()); - if (f.dataType() != v) { + if (!f.dataType().sameType(v)) { throw new IllegalArgumentException( String.format( "Column %s in input schema to write to "