Github user NicoK commented on a diff in the pull request: https://github.com/apache/flink/pull/4485#discussion_r138897087 --- Diff: flink-runtime/src/main/java/org/apache/flink/runtime/io/network/partition/consumer/SingleInputGate.java --- @@ -333,7 +333,7 @@ public void updateInputChannel(InputChannelDeploymentDescriptor icdd) throws IOE InputChannel current = inputChannels.get(partitionId); - if (current.getClass() == UnknownInputChannel.class) { + if (current instanceof UnknownInputChannel) { --- End diff -- Just to be on the safe side, you should also change this check in `#setInputChannel()` above. This way, we handle all sub-classes of `UnknownInputChannel` the same way as `UnknownInputChannel` itself
---