diff --git a/libp2p/pubsub/pubsub_notifee.py b/libp2p/pubsub/pubsub_notifee.py index 08b4a5b..b32c145 100644 --- a/libp2p/pubsub/pubsub_notifee.py +++ b/libp2p/pubsub/pubsub_notifee.py @@ -32,10 +32,10 @@ class PubsubNotifee(INotifee): self.dead_peers_queue = dead_peers_queue async def opened_stream(self, network: INetwork, stream: INetStream) -> None: - ... + await trio.hazmat.checkpoint() async def closed_stream(self, network: INetwork, stream: INetStream) -> None: - ... + await trio.hazmat.checkpoint() async def connected(self, network: INetwork, conn: INetConn) -> None: """ @@ -67,7 +67,7 @@ class PubsubNotifee(INotifee): ... async def listen(self, network: INetwork, multiaddr: Multiaddr) -> None: - ... + await trio.hazmat.checkpoint() async def listen_close(self, network: INetwork, multiaddr: Multiaddr) -> None: - ... + await trio.hazmat.checkpoint()