kernel-checkpoint 0.1.3__tar.gz → 0.1.5__tar.gz

{kernel_checkpoint-0.1.3 → kernel_checkpoint-0.1.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: kernel_checkpoint
-Version: 0.1.3
+Version: 0.1.5
 Summary: An extension for saving and restoring kernel pod state with jupyter enterprise gateway
 Project-URL: Homepage, https://github.com/lehuannhatrang/kernel-checkpoint-extension.git
 Project-URL: Bug Tracker, https://github.com/lehuannhatrang/kernel-checkpoint-extension.git/issues

{kernel_checkpoint-0.1.3 → kernel_checkpoint-0.1.5}/jupyter-notebook-image/Dockerfile

@@ -4,6 +4,6 @@ USER root
 
 RUN pip install --upgrade pip
 
-RUN pip install --no-cache-dir kernel-checkpoint==0.1.2
+RUN pip install --no-cache-dir kernel-checkpoint==0.1.5
 
 USER jovyan

{kernel_checkpoint-0.1.3 → kernel_checkpoint-0.1.5}/kernel_checkpoint/__init__.py

@@ -85,6 +85,7 @@ def _patch_kernel_handler_for_env_passthrough(server_app):
         server_app.config.GatewayClient.client_envs = [
             "KERNEL_CHECKPOINT_NAME",
             "KERNEL_CHECKPOINT_FILE_PATH",
+            "KERNEL_CHECKPOINT_CONTAINER_NAME",
         ]
     except Exception:
         pass

{kernel_checkpoint-0.1.3 → kernel_checkpoint-0.1.5}/kernel_checkpoint/_version.py

@@ -1,4 +1,4 @@
 # This file is auto-generated by Hatchling. As such, do not:
 #   - modify
 #   - track in version control e.g. be sure to add to .gitignore
-__version__ = VERSION = '0.1.3'
+__version__ = VERSION = '0.1.5'

{kernel_checkpoint-0.1.3 → kernel_checkpoint-0.1.5}/package.json

@@ -1,6 +1,6 @@
 {
     "name": "kernel-checkpoint",
-    "version": "0.1.3",
+    "version": "0.1.5",
     "description": "An extension for saving and restoring kernel pod state with jupyter enterprise gateway",
     "keywords": [
         "jupyter",

{kernel_checkpoint-0.1.3 → kernel_checkpoint-0.1.5}/src/checkpoint-panel.tsx

@@ -144,11 +144,11 @@ export function CheckpointPanel(props: ICheckpointPanelProps): JSX.Element {
     setFlash(null);
     try {
       const detail = await CheckpointAPI.getCheckpoint(namespace, checkpointName);
-      const filePath = detail.checkpointFiles?.[0]?.filePath ?? '';
-      if (!filePath) {
+      const checkpointFile = detail.checkpointFiles?.[0];
+      if (!checkpointFile?.storagePath) {
         throw new Error('No checkpoint file path available for this checkpoint');
       }
-      await onRestore(checkpointName, filePath);
+      await onRestore(checkpointName, checkpointFile.storagePath, checkpointFile.containerName);
       setFlash({
         type: 'success',
         text: `Kernel restored from "${checkpointName}". The kernel is restarting.`

{kernel_checkpoint-0.1.3 → kernel_checkpoint-0.1.5}/src/index.ts

@@ -92,7 +92,7 @@ const plugin: JupyterFrontEndPlugin<void> = {
         const kernelId = kernel.id;
         const kernelSpecName = kernel.name || 'python_kubernetes';
 
-        const onRestore = async (checkpointName: string, checkpointFilePath: string): Promise<void> => {
+        const onRestore = async (checkpointName: string, checkpointFilePath: string, containerName: string): Promise<void> => {
           const settings = ServerConnection.makeSettings();
           const kernelUrl = URLExt.join(settings.baseUrl, 'api', 'kernels');
 
@@ -105,7 +105,8 @@ const plugin: JupyterFrontEndPlugin<void> = {
                 name: kernelSpecName,
                 env: {
                   KERNEL_CHECKPOINT_NAME: checkpointName,
-                  KERNEL_CHECKPOINT_FILE_PATH: checkpointFilePath
+                  KERNEL_CHECKPOINT_FILE_PATH: checkpointFilePath,
+                  KERNEL_CHECKPOINT_CONTAINER_NAME: containerName
                 }
               })
             },

{kernel_checkpoint-0.1.3 → kernel_checkpoint-0.1.5}/src/types.ts

@@ -59,5 +59,5 @@ export interface ICheckpointPanelProps {
   namespace: string;
   kernelId: string;
   kernelSpecName: string;
-  onRestore: (checkpointName: string, checkpointFilePath: string) => Promise<void>;
+  onRestore: (checkpointName: string, checkpointFilePath: string, containerName: string) => Promise<void>;
 }