safe removal again

Netflix · Dec 17, 2024 · 84db134 · 84db134
1 parent ae2fee1
commit 84db134
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/metaflow/runner/click_api.py b/metaflow/runner/click_api.py
@@ -223,7 +223,11 @@ def extract_flow_class_from_file(flow_file: str) -> FlowSpec:
     finally:
         # Only remove from path if we added it
         if path_was_added:
-            sys.path.remove(flow_dir)
+            try:
+                sys.path.remove(flow_dir)
+            except ValueError:
+                # User's code might have removed it already
+                pass
 
 
 class MetaflowAPI(object):