sascha/workload-automation
1
0
Fork 0
mirror of https://github.com/ARM-software/workload-automation.git synced 2025-10-29 22:24:51 +00:00
Code Issues Releases Wiki Activity

Don't break prematurely when running a cell on an ipython kernel

Browse Source 
The kernel may go idle before it processes the next input, which break
the while=True loop in run_cell() early.  Wait for an acknowledgement
of the input we've sent to the kernel before considering an idle
message to mean that the cell has been parsed.
This commit is contained in:
Javi Merino
2015-04-17 16:18:23 +01:00
parent 0faa5ae455
commit 2b04cb38d9
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
wlauto/utils/ipython.py
View File

@@ -52,6 +52,7 @@ def run_cell(kernel_client, cell):
"""Run a cell of a notebook in an ipython kernel and return its output"""
kernel_client.execute(cell.input)
input_acknowledged = False
outs = []
while True:
msg = kernel_client.get_iopub_msg()
@@ -61,11 +62,12 @@ def run_cell(kernel_client, cell):
out = NotebookNode(output_type=msg_type)
if msg_type == "status":
if content["execution_state"] == "idle":
if content["execution_state"] == "idle" and input_acknowledged:
break
else:
continue
elif msg_type == "pyin":
input_acknowledged = True
continue
elif msg_type == "stream":
out.stream = content["name"]