diff --git a/pympipool/shared/taskexecutor.py b/pympipool/shared/taskexecutor.py index 78592a27..8ee7367d 100644 --- a/pympipool/shared/taskexecutor.py +++ b/pympipool/shared/taskexecutor.py @@ -42,6 +42,8 @@ def cloudpickle_register(ind=2): """ try: # When executed in a jupyter notebook this can cause a ValueError - in this case we just ignore it. cloudpickle.register_pickle_by_value(inspect.getmodule(inspect.stack()[ind][0])) + except IndexError: + cloudpickle_register(ind=ind - 1) except ValueError: pass