Skip to content

Commit 2c9fbf4

Browse files
authored
Merge pull request #740 from jtpio/await-kernel-ready
2 parents 2631730 + bf08cde commit 2c9fbf4

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

jupyter_client/multikernelmanager.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -260,10 +260,12 @@ async def _async_shutdown_kernel(
260260
if self._using_pending_kernels() and kernel_id in self._pending_kernels:
261261
raise RuntimeError("Kernel is in a pending state. Cannot shutdown.")
262262
# If the kernel is still starting, wait for it to be ready.
263-
elif kernel_id in self._starting_kernels:
264-
kernel = self._starting_kernels[kernel_id]
263+
elif kernel_id in self._pending_kernels:
264+
kernel = self._pending_kernels[kernel_id]
265265
try:
266266
await kernel
267+
km = self.get_kernel(kernel_id)
268+
await km.ready
267269
except Exception:
268270
self.remove_kernel(kernel_id)
269271
return

0 commit comments

Comments
 (0)