From 9cb00e79b279a46c2e22958ffae1ae9d2c8982c4 Mon Sep 17 00:00:00 2001 From: Alex Date: Sat, 18 May 2024 17:26:36 +0100 Subject: [PATCH] Fixing issues --- faststream/kafka/testing.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/faststream/kafka/testing.py b/faststream/kafka/testing.py index d835634179..798be0bbf6 100755 --- 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