Skip to content

Commit e71e958

Browse files
authored
Update kernel env to reflect changes in session (#1354)
1 parent 5c5b501 commit e71e958

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

jupyter_server/services/sessions/sessionmanager.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -461,6 +461,13 @@ async def update_session(self, session_id, **kwargs):
461461
query = "UPDATE session SET %s WHERE session_id=?" % (", ".join(sets)) # noqa
462462
self.cursor.execute(query, [*list(kwargs.values()), session_id])
463463

464+
if hasattr(self.kernel_manager, "update_env"):
465+
self.cursor.execute(
466+
"SELECT path, name, kernel_id FROM session WHERE session_id=?", [session_id]
467+
)
468+
path, name, kernel_id = self.cursor.fetchone()
469+
self.kernel_manager.update_env(kernel_id=kernel_id, env=self.get_kernel_env(path, name))
470+
464471
async def kernel_culled(self, kernel_id: str) -> bool:
465472
"""Checks if the kernel is still considered alive and returns true if its not found."""
466473
return kernel_id not in self.kernel_manager

0 commit comments

Comments
 (0)