chore: fix CI

airtai · Dec 16, 2024 · ab47f76 · ab47f76
1 parent 6e60761
commit ab47f76
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/tests/opentelemetry/kafka/test_kafka.py b/tests/opentelemetry/kafka/test_kafka.py
@@ -261,17 +261,19 @@ async def handler(m, baggage: CurrentBaggage):
 
 @pytest.mark.kafka
 class TestPublishWithTelemetry(TestPublish):
-    def get_broker(self, apply_types: bool = False):
+    def get_broker(self, apply_types: bool = False, **kwargs):
         return KafkaBroker(
             middlewares=(KafkaTelemetryMiddleware(),),
             apply_types=apply_types,
+            **kwargs,
         )
 
 
 @pytest.mark.kafka
 class TestConsumeWithTelemetry(TestConsume):
-    def get_broker(self, apply_types: bool = False):
+    def get_broker(self, apply_types: bool = False, **kwargs):
         return KafkaBroker(
             middlewares=(KafkaTelemetryMiddleware(),),
             apply_types=apply_types,
+            **kwargs,
         )