diff --git a/kafka/producer/kafka.py b/kafka/producer/kafka.py index ecaf4bc28..0c9a7e589 100644 --- a/kafka/producer/kafka.py +++ b/kafka/producer/kafka.py @@ -538,7 +538,8 @@ def _estimate_size_in_bytes(self, key, value, headers=[]): return LegacyRecordBatchBuilder.estimate_size_in_bytes( magic, self.config['compression_type'], key, value) - def send(self, topic, value=None, key=None, headers=None, partition=None, timestamp_ms=None) -> RecordMetadata: + def send(self, topic, value=None, key=None, headers=None, partition=None, timestamp_ms=None): + # type: (str, bytes, bytes, list, int, int) -> RecordMetadata """Publish a message to a topic. Arguments: