[BUG] Consider each site_package

Signed-off-by: Bennett <[email protected]>
flyteorg · Nov 12, 2024 · e48f56c · e48f56c
1 parent 3475ddc
commit e48f56c
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/flytekit/tools/script_mode.py b/flytekit/tools/script_mode.py
@@ -217,7 +217,9 @@ def list_imported_modules_as_files(source_path: str, modules: List[ModuleType])
             if os.path.commonpath([flytekit_root, mod_file]) == flytekit_root:
                 continue
 
-            if os.path.commonpath(site_packages + [mod_file]) in site_packages_set:
+            if any(
+                (os.path.commonpath([site_package, mod_file]) == site_package for site_package in site_packages_set)
+            ):
                 # Do not upload files from site-packages
                 continue