From 9c89411fd301be0b22d436d99fc953ebcf1379ed Mon Sep 17 00:00:00 2001
From: zt <zt@zjuici.com>
Date: Fri, 15 Nov 2024 15:15:23 +0800
Subject: [PATCH] feat: add `shutdown_all` for async and sync local kernels
---
src/pybox/local.py | 6 ++++++
1 file changed, 6 insertions(+)
diff --git a/src/pybox/local.py b/src/pybox/local.py
index 192f79e..b0f3641 100644
--- a/src/pybox/local.py
+++ b/src/pybox/local.py
@@ -304,3 +304,9 @@ async def ashutdown(
logger.warning("kernel %s not found", kernel_id)
else:
logger.info("Kernel %s shut down", kernel_id)
+
+ def shutdown_all(self):
+ self.kernel_manager.shutdown_all()
+
+ async def ashutdown_all(self):
+ await self.async_kernel_manager.shutdown_all()