From 2906f87089ec0ecccd87e6d1013d709db6e1f260 Mon Sep 17 00:00:00 2001 From: Jan Janssen Date: Thu, 3 Aug 2023 16:18:25 -0600 Subject: [PATCH 1/2] Add fall back option for cloudpickle_register() --- pympipool/shared/taskexecutor.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/pympipool/shared/taskexecutor.py b/pympipool/shared/taskexecutor.py index 78592a27..f75cbccd 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 From 1f286a44fbdc106864807fa053df746c40e14631 Mon Sep 17 00:00:00 2001 From: Jan Janssen Date: Thu, 3 Aug 2023 16:35:51 -0600 Subject: [PATCH 2/2] black formatting --- pympipool/shared/taskexecutor.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pympipool/shared/taskexecutor.py b/pympipool/shared/taskexecutor.py index f75cbccd..8ee7367d 100644 --- a/pympipool/shared/taskexecutor.py +++ b/pympipool/shared/taskexecutor.py @@ -43,7 +43,7 @@ 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) + cloudpickle_register(ind=ind - 1) except ValueError: pass