jupyterlab-ipyflow 0.0.180 → 0.0.181

package/lib/index.js

@@ -144,6 +144,16 @@ function computeTransitiveClosureHelper(closure, cellId, edges) {
     }
     children.forEach((child) => computeTransitiveClosureHelper(closure, child, edges));
 }
+function mergeMaps(priority, backup) {
+    const merged = {};
+    for (const key in backup) {
+        merged[key] = backup[key];
+    }
+    for (const key in priority) {
+        merged[key] = priority[key];
+    }
+    return merged;
+}
 /**
  * Initialization data for the jupyterlab-ipyflow extension.
  */
@@ -742,7 +752,7 @@ const connectToComm = (session, notebooks, notebook) => {
     };
     notebooks.selectionChanged.connect(onSelectionChanged);
     comm.onMsg = (msg) => {
-        var _a, _b;
+        var _a, _b, _c, _d;
         const payload = msg.content.data;
         if (disconnected || !((_a = payload.success) !== null && _a !== void 0 ? _a : true)) {
             return;
@@ -761,8 +771,11 @@ const connectToComm = (session, notebooks, notebook) => {
         }
         else if (payload.type === 'compute_exec_schedule') {
             state.settings = payload.settings;
-            state.cellParents = payload.cell_parents;
-            state.cellChildren = payload.cell_children;
+            const ipyflow_metadata = notebook.model.getMetadata('ipyflow');
+            const parentsFromMetadata = (_b = ipyflow_metadata === null || ipyflow_metadata === void 0 ? void 0 : ipyflow_metadata.cell_parents) !== null && _b !== void 0 ? _b : {};
+            const childrenFromMetadata = (_c = ipyflow_metadata === null || ipyflow_metadata === void 0 ? void 0 : ipyflow_metadata.cell_children) !== null && _c !== void 0 ? _c : {};
+            state.cellParents = mergeMaps(payload.cell_parents, parentsFromMetadata);
+            state.cellChildren = mergeMaps(payload.cell_children, childrenFromMetadata);
             notebook.model.setMetadata('ipyflow', {
                 cell_parents: state.cellParents,
                 cell_children: state.cellChildren,
@@ -790,7 +803,7 @@ const connectToComm = (session, notebooks, notebook) => {
             const exec_mode = payload.exec_mode;
             state.isReactivelyExecuting =
                 state.isReactivelyExecuting ||
-                    ((_b = payload === null || payload === void 0 ? void 0 : payload.is_reactively_executing) !== null && _b !== void 0 ? _b : false) ||
+                    ((_d = payload === null || payload === void 0 ? void 0 : payload.is_reactively_executing) !== null && _d !== void 0 ? _d : false) ||
                     exec_mode === 'reactive';
             if (exec_mode === 'reactive') {
                 state.newReadyCells = new Set([

package/package.json

@@ -67,5 +67,5 @@
     "extension": true,
     "outputDir": "../../core/ipyflow/resources/labextension/"
   },
-  "version": "0.0.180"
+  "version": "0.0.181"
 }