diff --git a/jupyter_client/asynchronous/client.py b/jupyter_client/asynchronous/client.py index 11873416..cde8ecaf 100644 --- a/jupyter_client/asynchronous/client.py +++ b/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 diff --git a/jupyter_client/channels.py b/jupyter_client/channels.py index c645b134..ea1f4cbf 100644 --- a/jupyter_client/channels.py +++ b/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: