diff --git a/src/mdevaluate/autosave.py b/src/mdevaluate/autosave.py index 3026e6b..d0e7d4e 100644 --- a/src/mdevaluate/autosave.py +++ b/src/mdevaluate/autosave.py @@ -166,8 +166,10 @@ def autosave_data( @functools.wraps(function) def autosave(*args, **kwargs): description = kwargs.pop("description", "") + autosave_dir_overwrite = kwargs.pop("autosave_dir_overwrite", None) + autosave_dir = autosave_dir_overwrite if autosave_dir_overwrite is not None else autosave_directory autoload = kwargs.pop("autoload", True) and load_autosave_data - if autosave_directory is not None: + if autosave_dir is not None: relevant_args = list(args[:nargs]) if kwargs_keys is not None: for key in [*posargs_keys, *kwargs_keys]: