diff --git a/pypots/base.py b/pypots/base.py index bb0a27e1..e444ec24 100644 --- a/pypots/base.py +++ b/pypots/base.py @@ -291,6 +291,7 @@ def save( f"❌ File {saving_path} exists. Saving operation aborted. " f"Use the arg `overwrite=True` to force overwrite." ) + return try: create_dir_if_not_exist(saving_dir) diff --git a/requirements/conda_env.yml b/requirements/conda_env.yml index 39c03e27..5ffb13b1 100644 --- a/requirements/conda_env.yml +++ b/requirements/conda_env.yml @@ -22,8 +22,5 @@ dependencies: - conda-forge::pygrinder >=0.6.4 - conda-forge::tsdb >=0.6.1 - conda-forge::benchpots >=0.3 + - conda-forge::ai4ts - pytorch::pytorch >=1.10.0 - - - pip: - # not published on conda-forge yet - - ai4ts \ No newline at end of file