try removing multiple mports to resolve fast-mode issue (#2110)

Signed-off-by: novahow <[email protected]> Co-authored-by: Kevin Su <[email protected]> Signed-off-by: Jan Fiedler <[email protected]>
flyteorg · Jul 25, 2024 · d8898bb · d8898bb
1 parent 635eb03
commit d8898bb
Showing 1 changed file with 0 additions and 4 deletions.
diff --git a/flytekit/core/tracker.py b/flytekit/core/tracker.py
@@ -16,11 +16,7 @@ def import_module_from_file(module_name, file):
     try:
         spec = importlib.util.spec_from_file_location(module_name, file)
         module = importlib.util.module_from_spec(spec)
-        spec.loader.exec_module(module)
         return module
-    except AssertionError:
-        # handle where we can't determine the module of functions within the module
-        return importlib.import_module(module_name)
     except Exception as exc:
         raise ModuleNotFoundError(f"Module from file {file} cannot be loaded") from exc