fix cyclic import

aio-libs · Jul 2, 2024 · 2be5d39 · 2be5d39
1 parent 254f907
commit 2be5d39
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/aiokafka/cluster.py b/aiokafka/cluster.py
@@ -7,6 +7,7 @@
 import time
 from concurrent.futures import Future
 from typing import (
+    TYPE_CHECKING,
     Any,
     Callable,
     Dict,
@@ -22,7 +23,6 @@
 from typing_extensions import TypeAlias, TypedDict
 
 from aiokafka import errors as Errors
-from aiokafka.client import CoordinationType
 from aiokafka.conn import collect_hosts
 from aiokafka.protocol.commit import (
     GroupCoordinatorResponse_v0,
@@ -38,6 +38,9 @@
 )
 from aiokafka.structs import BrokerMetadata, PartitionMetadata, TopicPartition
 
+if TYPE_CHECKING:
+    from aiokafka.client import CoordinationType
+
 ListenerCallable: TypeAlias = Callable[["ClusterMetadata"], None]
 NodeId: TypeAlias = Union[str, int]