Update flytekit/core/tracker.py

Co-authored-by: Yee Hing Tong <wild-endeavor@users.noreply.github.com>
flyteorg · Sep 25, 2024 · 5e4076d · 5e4076d
1 parent 0c2e60e
commit 5e4076d
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/flytekit/core/tracker.py b/flytekit/core/tracker.py
@@ -270,6 +270,8 @@ def _resolve_abs_module_name(self, path: str, package_root: typing.Optional[str]
         # Let us remove any extensions like .py
         basename = os.path.splitext(basename)[0]
 
+        # This is an escape hatch for the zipimporter (used by spark).  As this function is called recursively,
+        # it'll eventually reach the zip file, which is not extracted, so we should return.
         if not Path(dirname).is_dir():
             return basename