Fixed type annotations to support Python 2.7

wbarnha · Sep 22, 2022 · 0c64bc7 · 0c64bc7
1 parent 1882872
commit 0c64bc7
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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: