Fixing issues

airtai · May 18, 2024 · 9cb00e7 · 9cb00e7
1 parent a03f4ea
commit 9cb00e7
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/faststream/kafka/testing.py b/faststream/kafka/testing.py
@@ -1,5 +1,6 @@
 from datetime import datetime
-from typing import TYPE_CHECKING, Any, Dict, Optional
+from typing import TYPE_CHECKING, Any, Dict, Optional, Callable
+from unittest.mock import AsyncMock
 
 from aiokafka import ConsumerRecord
 from typing_extensions import override