diff --git a/libp2p/pubsub/pubsub_notifee.py b/libp2p/pubsub/pubsub_notifee.py index 2d66da8..be48190 100644 --- a/libp2p/pubsub/pubsub_notifee.py +++ b/libp2p/pubsub/pubsub_notifee.py @@ -1,6 +1,7 @@ from typing import TYPE_CHECKING from multiaddr import Multiaddr +import trio from libp2p.network.connection.net_connection_interface import INetConn from libp2p.network.network_interface import INetwork @@ -8,7 +9,6 @@ from libp2p.network.notifee_interface import INotifee from libp2p.network.stream.net_stream_interface import INetStream if TYPE_CHECKING: - import trio # noqa: F401 from libp2p.peer.id import ID # noqa: F401 @@ -30,9 +30,9 @@ class PubsubNotifee(INotifee): can process dead peers after we disconnect from each other """ self.initiator_peers_queue = initiator_peers_queue - self.initiator_peers_queue_lock: trio.Lock() + self.initiator_peers_queue_lock = trio.Lock() self.dead_peers_queue = dead_peers_queue - self.dead_peers_queue_lock: trio.Lock() + self.dead_peers_queue_lock = trio.Lock() async def opened_stream(self, network: INetwork, stream: INetStream) -> None: pass