diff --git a/mdevaluate/autosave.py b/mdevaluate/autosave.py index 6ad70e0..d387819 100644 --- a/mdevaluate/autosave.py +++ b/mdevaluate/autosave.py @@ -126,7 +126,7 @@ def save_data(filename, checksum, data): def load_data(filename): """Load data from a npz file.""" notify('Loading result from file: {}'.format(filename)) - fdata = np.load(filename) + fdata = np.load(filename, alloe_pickle=True) if 'data' in fdata: return fdata['data'] else: @@ -171,7 +171,7 @@ def autosave_data(nargs, kwargs_keys=None, version=None): filename = get_filename(function, csum, description, *relevant_args) if autoload and (verify_file(filename, csum) or verify_file(filename, legacy_csum)): - result = load_data(filename, allow_pickle) + result = load_data(filename) else: result = function(*args, **kwargs) save_data(filename, csum, result)