From 2c75d3418bcb8527cc22bcbbfae52af0fdb45dfd Mon Sep 17 00:00:00 2001 From: spataphore1337 Date: Sun, 5 May 2024 14:01:47 +0300 Subject: [PATCH] add partitions for __init__ of Default and Batch Subscriber --- faststream/kafka/subscriber/usecase.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/faststream/kafka/subscriber/usecase.py b/faststream/kafka/subscriber/usecase.py index 5af13fdd8b..7b0f34b76a 100644 --- a/faststream/kafka/subscriber/usecase.py +++ b/faststream/kafka/subscriber/usecase.py @@ -267,6 +267,7 @@ def __init__( group_id: Optional[str], listener: Optional["ConsumerRebalanceListener"], pattern: Optional[str], + partitions: Optional[Iterable["TopicPartition"]], builder: Callable[..., "AIOKafkaConsumer"], is_manual: bool, # Subscriber args @@ -284,6 +285,7 @@ def __init__( group_id=group_id, listener=listener, pattern=pattern, + partitions=partitions, builder=builder, is_manual=is_manual, # subscriber args @@ -315,6 +317,7 @@ def __init__( group_id: Optional[str], listener: Optional["ConsumerRebalanceListener"], pattern: Optional[str], + partitions: Optional[Iterable["TopicPartition"]], builder: Callable[..., "AIOKafkaConsumer"], is_manual: bool, # Subscriber args @@ -337,6 +340,7 @@ def __init__( group_id=group_id, listener=listener, pattern=pattern, + partitions=partitions, builder=builder, is_manual=is_manual, # subscriber args