Fix for env_escape bug when importing local packages (#1100)

Netflix · Aug 5, 2022 · eb8b373 · eb8b373
1 parent 6cb54b5
commit eb8b373
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/metaflow/plugins/env_escape/__init__.py b/metaflow/plugins/env_escape/__init__.py
@@ -134,6 +134,7 @@ def load():
                 print("Not using environment escape for '%s' as module present" % prefix)
             # In both cases, we don't load our loader since
             # the package is locally present
+            sys.path = old_paths
             return
     sys.path = old_paths
     m = ModuleImporter("{python_executable}", "{pythonpath}", {max_pickle_version}, "{path}", {prefixes})