@Override
    public Set<TopicPartition> assignment() {
        if (consumer == null) {
            throw new IllegalWorkerStateException("SinkTaskContext may not be used to look up partition assignment until the task is initialized");
        }
        return consumer.assignment();
    }
Get the timeout in milliseconds set by SinkTasks. Used by the Kafka Connect framework.
@return the backoff timeout in milliseconds.
connect/runtime/src/main/java/org/apache/kafka/connect/runtime/WorkerSinkTaskContext.java