conkernelclient 0.0.9__tar.gz → 0.0.10__tar.gz

{conkernelclient-0.0.9/conkernelclient.egg-info → conkernelclient-0.0.10}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: conkernelclient
-Version: 0.0.9
+Version: 0.0.10
 Summary: Concurrent-safe Jupyter KernelClient
 Author-email: Jeremy Howard <github@jhoward.fastmail.fm>
 License: Apache-2.0

conkernelclient-0.0.10/conkernelclient/__init__.py

@@ -0,0 +1,2 @@
+__version__ = "0.0.10"
+from .core import *

{conkernelclient-0.0.9 → conkernelclient-0.0.10}/conkernelclient/_modidx.py

@@ -21,4 +21,5 @@ d = { 'settings': { 'branch': 'main',
                                       'conkernelclient.core.ConKernelClient.stop_channels': ( 'core.html#conkernelclient.stop_channels',
                                                                                               'conkernelclient/core.py'),
                                       'conkernelclient.core.ConKernelManager': ('core.html#conkernelmanager', 'conkernelclient/core.py'),
+                                      'conkernelclient.core.DeadKernelError': ('core.html#deadkernelerror', 'conkernelclient/core.py'),
                                       'conkernelclient.core._send': ('core.html#_send', 'conkernelclient/core.py')}}}

{conkernelclient-0.0.9 → conkernelclient-0.0.10}/conkernelclient/core.py

@@ -3,7 +3,7 @@
 # AUTOGENERATED! DO NOT EDIT! File to edit: ../nbs/00_core.ipynb.
 
 # %% auto #0
-__all__ = ['ConKernelClient', 'ConKernelManager']
+__all__ = ['DeadKernelError', 'ConKernelClient', 'ConKernelManager']
 
 # %% ../nbs/00_core.ipynb #d3b521c0
 from queue import Empty
@@ -20,6 +20,9 @@ import asyncio, zmq.asyncio, time, logging
 # %% ../nbs/00_core.ipynb #737a0fc1
 _log = logging.getLogger(__name__)
 
+# %% ../nbs/00_core.ipynb #a093006c
+class DeadKernelError(Exception): pass
+
 # %% ../nbs/00_core.ipynb #374b75d0
 if not hasattr(Session, '_orig_send'): Session._orig_send = Session.send
 
@@ -61,6 +64,7 @@ class ConKernelClient(AsyncKernelClient):
             while True:
                 try: reply = await self.get_shell_msg(timeout=None)
                 except Exception as e:
+                    if isinstance(e, (AssertionError, OSError, ZMQError)): e = DeadKernelError(f"Kernel socket closed: {e}")
                     self._fail_pending(e)
                     _log.warning(f"_reader died, pending={list(self._pending)}: {e}")
                     break
@@ -111,13 +115,15 @@ class ConKernelClient(AsyncKernelClient):
         if subsh_id is not None: msg["header"]["subshell_id"] = subsh_id
         msg_id = msg["header"]["msg_id"]
         if reply: self._pending[msg_id] = (asyncio.Queue(maxsize=1), stop_on_error)
-        self.shell_channel.send(msg)
+        try: self.shell_channel.send(msg)
+        except (AssertionError, OSError, ZMQError) as e: raise DeadKernelError(f"Kernel socket closed: {e}") from e
         if not reply: return msg_id
         return self._async_recv_reply(msg_id, timeout=timeout)
 
     async def stdin_send(self, msg:dict):
         "Send on `stdin_channel` then ping event loop"
-        self.stdin_channel.send(msg)
+        try: self.stdin_channel.send(msg)
+        except (AssertionError, OSError, ZMQError) as e: raise DeadKernelError(f"Kernel socket closed: {e}") from e
         await asyncio.sleep(0.01)
 
 # %% ../nbs/00_core.ipynb #b828c222

{conkernelclient-0.0.9 → conkernelclient-0.0.10/conkernelclient.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: conkernelclient
-Version: 0.0.9
+Version: 0.0.10
 Summary: Concurrent-safe Jupyter KernelClient
 Author-email: Jeremy Howard <github@jhoward.fastmail.fm>
 License: Apache-2.0

conkernelclient-0.0.9/conkernelclient/__init__.py

@@ -1,2 +0,0 @@
-__version__ = "0.0.9"
-from .core import *