perf(Ensemble): allow dumping of ensemble output into already existin…

…g directories
frankkramer-lab · Jun 13, 2022 · ab6c0bc · ab6c0bc
1 parent ea04aad
commit ab6c0bc
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 6 deletions.
diff --git a/aucmedi/ensemble/bagging.py b/aucmedi/ensemble/bagging.py
@@ -365,11 +365,12 @@ def dump(self, directory_path):
         if self.cache_dir is None:
             raise FileNotFoundError("Bagging does not have a valid model cache directory!")
         elif isinstance(self.cache_dir, tempfile.TemporaryDirectory):
-            shutil.copytree(self.cache_dir.name, directory_path)
+            shutil.copytree(self.cache_dir.name, directory_path,
+                            dirs_exist_ok=True)
             self.cache_dir.cleanup()
             self.cache_dir = directory_path
         else:
-            shutil.copytree(self.cache_dir, directory_path)
+            shutil.copytree(self.cache_dir, directory_path, dirs_exist_ok=True)
             self.cache_dir = directory_path
 
     # Load model from file

diff --git a/aucmedi/ensemble/composite.py b/aucmedi/ensemble/composite.py
@@ -525,11 +525,12 @@ def dump(self, directory_path):
         if self.cache_dir is None:
             raise FileNotFoundError("Composite does not have a valid model cache directory!")
         elif isinstance(self.cache_dir, tempfile.TemporaryDirectory):
-            shutil.copytree(self.cache_dir.name, directory_path)
+            shutil.copytree(self.cache_dir.name, directory_path,
+                            dirs_exist_ok=True)
             self.cache_dir.cleanup()
             self.cache_dir = directory_path
         else:
-            shutil.copytree(self.cache_dir, directory_path)
+            shutil.copytree(self.cache_dir, directory_path, dirs_exist_ok=True)
             self.cache_dir = directory_path
 
     # Load model from file

diff --git a/aucmedi/ensemble/stacking.py b/aucmedi/ensemble/stacking.py
@@ -514,11 +514,12 @@ def dump(self, directory_path):
         if self.cache_dir is None:
             raise FileNotFoundError("Stacking does not have a valid model cache directory!")
         elif isinstance(self.cache_dir, tempfile.TemporaryDirectory):
-            shutil.copytree(self.cache_dir.name, directory_path)
+            shutil.copytree(self.cache_dir.name, directory_path,
+                            dirs_exist_ok=True)
             self.cache_dir.cleanup()
             self.cache_dir = directory_path
         else:
-            shutil.copytree(self.cache_dir, directory_path)
+            shutil.copytree(self.cache_dir, directory_path, dirs_exist_ok=True)
             self.cache_dir = directory_path
 
     # Load model from file