fix: delete temporary folders locally only if they exist

phueb · Mar 21, 2021 · 85ffb2d · 85ffb2d
1 parent f05b4b6
commit 85ffb2d
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/ludwig/__init__.py b/ludwig/__init__.py
@@ -1,6 +1,6 @@
 
 
-__version__ = '4.0.1'
+__version__ = '4.0.2'
 
 
 def print_ludwig(s):

diff --git a/ludwig/__main__.py b/ludwig/__main__.py
@@ -263,8 +263,9 @@ def submit():
 
                 # temporary runs folder auto-created with name = {project_name}_runs must be removed
                 path_tmp = cwd / f'{src_path.name}_runs'
-                shutil.rmtree(path_tmp)
-                print(f'Removed temporary directory {path_tmp}')
+                if path_tmp.exists():
+                    shutil.rmtree(path_tmp)
+                    print(f'Removed temporary directory {path_tmp}')
 
             # if running on Ludwig, save worker instructions to shared drive
             else: