Fix reject 'Control comm was closed too early' when comm closed as ex…

…pected.
jupyter-widgets · Mar 10, 2024 · 200edee · 200edee
1 parent ae6df72
commit 200edee
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/packages/base-manager/src/manager-base.ts b/packages/base-manager/src/manager-base.ts
@@ -413,7 +413,10 @@ export abstract class ManagerBase implements IWidgetManager {
           resolve(null);
         });
 
-        initComm.on_close(() => reject('Control comm was closed too early'));
+        initComm.on_close(() => {
+          if (data.method !== 'update_states')
+            reject('Control comm was closed too early');
+        });
 
         // Send a states request msg
         initComm.send({ method: 'request_states' }, {});