diff --git a/tests/test_asyncio.py b/tests/test_asyncio.py index e22f9d36e..e2ec0c82c 100644 --- a/tests/test_asyncio.py +++ b/tests/test_asyncio.py @@ -11,12 +11,12 @@ from multiprocessing import Process import pytest +from anyio import create_task_group, move_on_after, sleep +from anyioutils import CancelledError, create_task from pytest import mark import zmq import zmq.asyncio as zaio -from anyio import create_task_group, move_on_after, sleep -from anyioutils import CancelledError, create_task pytestmark = pytest.mark.anyio diff --git a/zmq/_future.py b/zmq/_future.py index d14f309b3..51edaf13a 100644 --- a/zmq/_future.py +++ b/zmq/_future.py @@ -105,9 +105,13 @@ def _clear_wrapper_io(f): socket = self._socket_class.from_socket(socket) wrapped_sockets.append(socket) if mask & _zmq.POLLIN: - create_task(socket._add_recv_event(tg, 'poll', future=watcher), tg) + create_task( + socket._add_recv_event(tg, 'poll', future=watcher), tg + ) if mask & _zmq.POLLOUT: - create_task(socket._add_send_event(tg, 'poll', future=watcher), tg) + create_task( + socket._add_send_event(tg, 'poll', future=watcher), tg + ) else: raw_sockets.append(socket) evt = 0