chore: fix CI

airtai · Dec 16, 2024 · 6e60761 · 6e60761
1 parent e4713df
commit 6e60761
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/docs/docs/en/public_api b/docs/docs/en/public_api
@@ -1 +1 @@
-./api/
+api
diff --git a/tests/brokers/kafka/test_publish.py b/tests/brokers/kafka/test_publish.py
@@ -1,4 +1,5 @@
 import asyncio
+from typing import Any
 from unittest.mock import Mock
 
 import pytest
@@ -11,8 +12,8 @@
 
 @pytest.mark.kafka
 class TestPublish(BrokerPublishTestcase):
-    def get_broker(self, apply_types: bool = False):
-        return KafkaBroker(apply_types=apply_types)
+    def get_broker(self, apply_types: bool = False, **kwargs: Any) -> KafkaBroker:
+        return KafkaBroker(apply_types=apply_types, **kwargs)
 
     @pytest.mark.asyncio
     async def test_publish_batch(self, queue: str):