Allow also capture output from SoS cell #32

vatlab · Jan 26, 2018 · 3459dbf · 3459dbf
1 parent c85849e
commit 3459dbf
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/sos_notebook/kernel.py b/src/sos_notebook/kernel.py
@@ -76,6 +76,8 @@ def write(self, content):
                                               f'<div class="sos_hint">{content[10:].strip()}</div>').data}
                                       })
         else:
+            if self.name == 'stdout' and self.kernel._capture_result is not None:
+                self.kernel._capture_result += content
             self.kernel.send_response(self.kernel.iopub_socket, 'stream',
                                       {'name': self.name, 'text': content})