From f6badadb47ce2ee36d7269b7ad38400e7bc5404a Mon Sep 17 00:00:00 2001 From: Vidar Tonaas Fauske Date: Fri, 5 Feb 2021 12:08:46 +0000 Subject: [PATCH] Update MappingKM.restart_kernel to accept now kwarg This should bring it in line with the signature of the base class (added in jupyter_client 4.0.0). --- jupyter_server/services/kernels/kernelmanager.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/jupyter_server/services/kernels/kernelmanager.py b/jupyter_server/services/kernels/kernelmanager.py index f6f9cc5064..bb82ee15fc 100644 --- a/jupyter_server/services/kernels/kernelmanager.py +++ b/jupyter_server/services/kernels/kernelmanager.py @@ -315,10 +315,10 @@ def shutdown_kernel(self, kernel_id, now=False, restart=False): return self.pinned_superclass.shutdown_kernel(self, kernel_id, now=now, restart=restart) - async def restart_kernel(self, kernel_id): + async def restart_kernel(self, kernel_id, now=False): """Restart a kernel by kernel_id""" self._check_kernel_id(kernel_id) - await ensure_async(self.pinned_superclass.restart_kernel(self, kernel_id)) + await ensure_async(self.pinned_superclass.restart_kernel(self, kernel_id, now=now)) kernel = self.get_kernel(kernel_id) # return a Future that will resolve when the kernel has successfully restarted channel = kernel.connect_shell()