add partitions for __init__ of Default and Batch Subscriber

airtai · May 5, 2024 · 2c75d34 · 2c75d34
1 parent 8a4a562
commit 2c75d34
Showing 1 changed file with 4 additions and 0 deletions.
diff --git 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