Merge pull request #518 from WenjieDu/dev

Fix unintended overwrite when saving models and update conda dependencies
WenjieDu · Sep 20, 2024 · 7f12620 · 7f12620
2 parents 0879253 + 9c18cdc
commit 7f12620
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git 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
@@ -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