Fix mypy

jupyter · May 14, 2024 · a563aa6 · a563aa6
1 parent 41479de
commit a563aa6
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -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: