Merge pull request #740 from jtpio/await-kernel-ready

jupyter · Jan 21, 2022 · 2c9fbf4 · 2c9fbf4
2 parents 2631730 + bf08cde
commit 2c9fbf4
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/jupyter_client/multikernelmanager.py b/jupyter_client/multikernelmanager.py
@@ -260,10 +260,12 @@ async def _async_shutdown_kernel(
         if self._using_pending_kernels() and kernel_id in self._pending_kernels:
             raise RuntimeError("Kernel is in a pending state. Cannot shutdown.")
         # If the kernel is still starting, wait for it to be ready.
-        elif kernel_id in self._starting_kernels:
-            kernel = self._starting_kernels[kernel_id]
+        elif kernel_id in self._pending_kernels:
+            kernel = self._pending_kernels[kernel_id]
             try:
                 await kernel
+                km = self.get_kernel(kernel_id)
+                await km.ready
             except Exception:
                 self.remove_kernel(kernel_id)
                 return