Skip to content

Commit ca11207

Browse files
Modify the assert to accept any instance of AvroCoder, including subclasses. (#34843) (#34850)
Co-authored-by: fozzie15 <[email protected]>
1 parent 998de4f commit ca11207

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

sdks/java/io/kafka/src/main/java/org/apache/beam/sdk/io/kafka/KafkaReadSchemaTransformProvider.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -196,7 +196,7 @@ public PCollectionRowTuple expand(PCollectionRowTuple input) {
196196
PCollection<GenericRecord> kafkaValues =
197197
input.getPipeline().apply(kafkaRead.withoutMetadata()).apply(Values.create());
198198

199-
assert kafkaValues.getCoder().getClass() == AvroCoder.class;
199+
assert kafkaValues.getCoder() instanceof AvroCoder;
200200
AvroCoder<GenericRecord> coder = (AvroCoder<GenericRecord>) kafkaValues.getCoder();
201201
kafkaValues = kafkaValues.setCoder(AvroUtils.schemaCoder(coder.getSchema()));
202202
return PCollectionRowTuple.of("output", kafkaValues.apply(Convert.toRows()));

0 commit comments

Comments
 (0)