Propagate future out for sending status messages

jupyter · Nov 5, 2018 · c3d63d0 · c3d63d0
1 parent 0a18fb9
commit c3d63d0
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/notebook/services/kernels/handlers.py b/notebook/services/kernels/handlers.py
@@ -396,19 +396,19 @@ def _send_status_message(self, status):
             {'execution_state': status}
         )
         ws_msg = serialize_message(msg, channel='iopub')
-        self.write_message(ws_msg, binary=isinstance(ws_msg, bytes))
+        return self.write_message(ws_msg, binary=isinstance(ws_msg, bytes))
 
     def on_kernel_died(self, _data):
         logging.warning("kernel %s died, noticed by auto restarter", self.kernel_id)
-        self._send_status_message('restarting')
+        return self._send_status_message('restarting')
 
     def on_kernel_restarted(self, _data):
         logging.warning("kernel %s restarted", self.kernel_id)
-        self._send_status_message('starting')
+        return self._send_status_message('starting')
 
     def on_restart_failed(self, _data):
         logging.error("kernel %s restarted failed!", self.kernel_id)
-        self._send_status_message('dead')
+        return self._send_status_message('dead')
 
 
 #-----------------------------------------------------------------------------