diff --git a/src/abel/classes/runnable.py b/src/abel/classes/runnable.py index c82497e4..fa0e554b 100644 --- a/src/abel/classes/runnable.py +++ b/src/abel/classes/runnable.py @@ -154,9 +154,11 @@ def save(self): pickle.dump(self, savefile) # load object from file - def load(self, shot=None): + def load(self, shot=None, obj_path=None): import dill as pickle - with open(self.object_path(shot), 'rb') as loadfile: + if obj_path is None: + obj_path = self.object_path(shot) + with open(obj_path, 'rb') as loadfile: try: obj = pickle.load(loadfile) return obj