use correct callback to ensure UI gets refreshed after comm connectio…

…n established
ipyflow · Jan 18, 2024 · deae523 · deae523
1 parent fec05d8
commit deae523
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/frontend/labextension/src/index.ts b/frontend/labextension/src/index.ts
@@ -1003,8 +1003,6 @@ const connectToComm = (
       return;
     }
 
-    notifyActiveCell(state.activeCell.model);
-
     if (state.dirtyCells.has(state.activeCell.model.id)) {
       (state.activeCell.model as any)._setDirty?.(true);
     }
@@ -1192,7 +1190,7 @@ const connectToComm = (
       refreshNodeMapping(notebook);
       notebook.activeCellChanged.connect(onActiveCellChange);
       notebook.activeCell.model.stateChanged.connect(onExecution);
-      notifyActiveCell(notebook.activeCell.model);
+      onActiveCellChange(notebook, notebook.activeCell);
       notebook.model.contentChanged.connect(onContentChanged);
       notebook.model.cells.changed.connect(onContentChanged);
       state.requestComputeExecSchedule();