conkernelclient 0.0.18__tar.gz → 0.0.19__tar.gz

{conkernelclient-0.0.18/conkernelclient.egg-info → conkernelclient-0.0.19}/PKG-INFO

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

{conkernelclient-0.0.18 → conkernelclient-0.0.19}/conkernelclient/__init__.py

@@ -1,3 +1,3 @@
-__version__ = "0.0.18"
+__version__ = "0.0.19"
 from .core import *
 from .ops import *

{conkernelclient-0.0.18 → conkernelclient-0.0.19}/conkernelclient/_modidx.py

@@ -27,6 +27,8 @@ d = { 'settings': { 'branch': 'main',
                                       'conkernelclient.core.ConKernelClient.wait_for_ready': ( 'core.html#conkernelclient.wait_for_ready',
                                                                                                'conkernelclient/core.py'),
                                       'conkernelclient.core.ConKernelManager': ('core.html#conkernelmanager', 'conkernelclient/core.py'),
+                                      'conkernelclient.core.ConKernelManager.__init__': ( 'core.html#conkernelmanager.__init__',
+                                                                                          'conkernelclient/core.py'),
                                       'conkernelclient.core.ConKernelManager._transport_encryption_default': ( 'core.html#conkernelmanager._transport_encryption_default',
                                                                                                                'conkernelclient/core.py'),
                                       'conkernelclient.core.DeadKernelError': ('core.html#deadkernelerror', 'conkernelclient/core.py'),

{conkernelclient-0.0.18 → conkernelclient-0.0.19}/conkernelclient/core.py

@@ -186,5 +186,8 @@ async def astop_channels(self:ConKernelClient):
 # %% ../nbs/00_core.ipynb #b828c222
 class ConKernelManager(AsyncKernelManager):
     client_class,client_factory = ConKernelClient,Type(ConKernelClient)
+    def __init__(self, *args, kernel_spec=None, **kwargs):
+        super().__init__(*args, **kwargs)
+        if kernel_spec is not None: self._kernel_spec = kernel_spec
     @default('transport_encryption')
     def _transport_encryption_default(self): return 'auto'

{conkernelclient-0.0.18 → conkernelclient-0.0.19}/conkernelclient/ops.py

@@ -14,6 +14,7 @@ from fastcore.ansi import strip_ansi
 from fastcore.nbio import msgs2outs, preferred_out
 from .core import ConKernelClient, ConKernelManager, DeadKernelError
 from jupywire.ops import parent_id, iopub_msgs, output_types
+from jupyter_client.kernelspec import KernelSpec
 _all_ = ['parent_id', 'iopub_msgs', 'output_types']
 from contextlib import asynccontextmanager
 from ast import literal_eval
@@ -202,12 +203,15 @@ async def interrupt(self:ConKernelClient, timeout=5):
 # %% ../nbs/01_ops.ipynb #edaf42c9
 @asynccontextmanager
 async def run_kernel(
-    kernel_name='python3', # Kernelspec name to launch
+    kernel_name='python3', # Kernelspec name to launch, or display name when `argv` is given
+    argv=None, # Kernel argv; bypasses kernelspec discovery when provided
     manager_cls=ConKernelManager, # Manager class, e.g. a subclass customizing launch
     **kwargs # Passed to `start_kernel` (e.g. `env`, `cwd`)
 ):
-    "Start a kernel, yield `(km, kc)` with channels running, and always shut both down"
-    km = manager_cls(kernel_name=kernel_name)
+    "Start a named or direct-argv kernel, yielding `(km, kc)`, and always shut it down"
+    manager_kwargs = dict(kernel_name=kernel_name)
+    if argv is not None: manager_kwargs['kernel_spec'] = KernelSpec(argv=argv, display_name=kernel_name)
+    km = manager_cls(**manager_kwargs)
     await km.start_kernel(**kwargs)
     kc = await km.client().start_channels()
     try: yield km, kc

{conkernelclient-0.0.18 → conkernelclient-0.0.19/conkernelclient.egg-info}/PKG-INFO

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

{conkernelclient-0.0.18 → conkernelclient-0.0.19}/tests/test_races.py

@@ -37,11 +37,10 @@ class LockedClient(ConKernelClient):
 class LockedKernelManager(ConKernelManager):
     client_class,client_factory = LockedClient,Type(LockedClient)
 
-
 def test_burst_replies_under_send_jitter():
     "20 pending replies at once, every send jittered: the reader must route each reply to its waiter."
     async def _run():
-        async with run_kernel(kernel_name="ipymini", manager_cls=LockedKernelManager) as (km, kc):
+        async with run_kernel("ipymini", ["ipymini", "-f", "{connection_file}"], manager_cls=LockedKernelManager) as (km, kc):
             reps = [kc.execute(f"x{i} = {i}; x{i}", reply=True, timeout=30) for i in range(20)]
             for i, r in enumerate(await asyncio.gather(*reps)): assert r["content"]["status"] == "ok", (i, r["content"])
     asyncio.run(_run())