only flush stream in dispatch_shell if possible

ipython · May 26, 2019 · c3f6e61 · c3f6e61
1 parent b81632e
commit c3f6e61
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/ipykernel/kernelbase.py b/ipykernel/kernelbase.py
@@ -245,7 +245,10 @@ def dispatch_shell(self, stream, msg):
             self._publish_status(u'idle')
             # flush to ensure reply is sent before
             # handling the next request
-            stream.flush(zmq.POLLOUT)
+            try:
+                stream.flush(zmq.POLLOUT)
+            except AttributeError:
+                pass
             return
 
         msg_type = msg['header']['msg_type']
@@ -283,7 +286,10 @@ def dispatch_shell(self, stream, msg):
         self._publish_status(u'idle')
         # flush to ensure reply is sent before
         # handling the next request
-        stream.flush(zmq.POLLOUT)
+        try:
+            stream.flush(zmq.POLLOUT)
+        except AttributeError:
+            pass
 
     def pre_handler_hook(self):
         """Hook to execute before calling message handler"""