Errors shown under wrong cell using magics

[piotrszajer]

Hi, I'm facing an issue where Traceback is shown under first cell, where IPyParallel is defined, not under the actual cell in which error was raised.

Steps to reproduce:

1. Terminal 1

pip install jupyter ipyparallel
ipcluster nbextension enable
jupyter server extension enable ipyparallel
ipcluster start --n=1

2. Terminal 2

jupyter notebook

3. Create new notebook
4. Cell 1: import this -> works fine
5. Cell 2:

import ipyparallel as ipp
rc = ipp.Client()
dview = rc[:]

-> works fine 6. Cell 3:

%%px
import os
print(f"Hello from engine {os.getpid()}")

-> works fine 7. Cell 4:

1/0

-> Traceback is printed under cell 2

What's more, further execution of cell 4 results in adding Traceback under cell 2, when cell 2 is re-run then Traceback stopped to show.

BugReproduce.zip

Additional: %pxresult shows Traceback in cell were it was executed

Any idea how to overcome it?