fix: handle load_session error

jupyter-server · Jun 12, 2024 · 3f14615 · 3f14615
1 parent d01e84a
commit 3f14615
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/enterprise_gateway/services/kernels/remotemanager.py b/enterprise_gateway/services/kernels/remotemanager.py
@@ -197,7 +197,11 @@ def check_kernel_id(self, kernel_id: str) -> None:
 
     def _refresh_kernel(self, kernel_id: str) -> bool:
         if self.parent.availability_mode == EnterpriseGatewayConfigMixin.AVAILABILITY_REPLICATION:
-            self.parent.kernel_session_manager.load_session(kernel_id)
+            try:
+                self.parent.kernel_session_manager.load_session(kernel_id)
+            except Exception as e:
+                self.log.error(f"Failed to load session, kernel_id:{kernel_id}", e)
+                return False
             return self.parent.kernel_session_manager.start_session(kernel_id)
         # else we should throw 404 when not using an availability mode of 'replication'
         return False