wip

jupyter-book · Feb 2, 2024 · abaa021 · abaa021
1 parent c1a9566
commit abaa021
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 0 deletions.
diff --git a/packages/myst-cli/src/build/build.ts b/packages/myst-cli/src/build/build.ts
@@ -186,4 +186,6 @@ export async function build(session: ISession, files: string[], opts: BuildOpts)
       await buildSite(session, opts);
     }
   }
+
+  session.maybeDisposeJupyterSessionManager();
 }
diff --git a/packages/myst-cli/src/session/session.ts b/packages/myst-cli/src/session/session.ts
@@ -75,6 +75,7 @@ export class Session implements ISession {
   _shownUpgrade = false;
   _latestVersion?: string;
   _jupyterSessionManager: SessionManager | undefined | null = null;
+  _disposeJupyterSessionManager?: () => void;
 
   get log(): Logger {
     return this.$logger;
@@ -207,4 +208,12 @@ export class Session implements ISession {
       return undefined;
     }
   }
+
+  maybeDisposeJupyterSessionManager() {
+    if (this._disposeJupyterSessionManager === undefined) {
+      return;
+    };
+    this._disposeJupyterSessionManager();
+    this._disposeJupyterSessionManager = undefined;
+  }
 }
diff --git a/packages/myst-cli/src/session/types.ts b/packages/myst-cli/src/session/types.ts
@@ -26,6 +26,7 @@ export type ISession = {
   loadPlugins(): Promise<MystPlugin>;
   getAllWarnings(ruleId: RuleId): (BuildWarning & { file: string })[];
   jupyterSessionManager(): Promise<SessionManager | undefined>;
+  maybeDisposeJupyterSessionManager(): void;
 };
 
 export type ISessionWithCache = ISession & {