Use non-blocking zmq Poller

jupyter_client/asynchronous/client.py

@@ -33,7 +33,7 @@ class AsyncKernelClient(KernelClient):
     raising :exc:`queue.Empty` if no message arrives within ``timeout`` seconds.
     """
 
-    context = Instance(zmq.asyncio.Context)
+    context = Instance(zmq.asyncio.Context)  # type:ignore[arg-type]
 
     def _context_default(self) -> zmq.asyncio.Context:
         self._created_context = True

jupyter_client/channels.py

@@ -220,7 +220,7 @@ def _recv(self, **kwargs: t.Any) -> t.Dict[str, t.Any]:
         ident, smsg = self.session.feed_identities(msg)
         return self.session.deserialize(smsg)
 
-    def get_msg(self, timeout: t.Optional[float] = None) -> t.Dict[str, t.Any]:
+    def get_msg(self, timeout: t.Optional[int] = None) -> t.Dict[str, t.Any]:
         """Gets a message if there is one that is ready."""
         assert self.socket is not None
         if timeout is not None:

jupyter_client/client.py

@@ -530,7 +530,7 @@ async def _async_execute_interactive(
         else:
             timeout_ms = None
 
-        poller = zmq.Poller()
+        poller = zmq.asyncio.Poller()
         iopub_socket = self.iopub_channel.socket
         poller.register(iopub_socket, zmq.POLLIN)
         if allow_stdin:
@@ -544,7 +544,7 @@ async def _async_execute_interactive(
             if timeout is not None:
                 timeout = max(0, deadline - time.monotonic())
                 timeout_ms = int(1000 * timeout)
-            events = dict(poller.poll(timeout_ms))
+            events = dict(await poller.poll(timeout_ms))
             if not events:
                 emsg = "Timeout waiting for output"
                 raise TimeoutError(emsg)

pyproject.toml

@@ -51,7 +51,7 @@ test = [
     "mypy",
     "paramiko; sys_platform == 'win32'",
     "pre-commit",
-    "pytest",
+    "pytest<8.2.0",
     "pytest-jupyter[client]>=0.4.1",
     "pytest-cov",
     "pytest-timeout",