Supports importing modules in current path (#2830)

Signed-off-by: Ketan Umare <[email protected]> Co-authored-by: Ketan Umare <[email protected]>
flyteorg · Nov 8, 2024 · 9d8861b · 9d8861b
1 parent b7c7c13
commit 9d8861b
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/flytekit/interaction/click_types.py b/flytekit/interaction/click_types.py
@@ -6,6 +6,7 @@
 import logging
 import os
 import pathlib
+import sys
 import typing
 import typing as t
 from typing import cast, get_args
@@ -149,6 +150,7 @@ def convert(
                 click.echo(f"Did not receive a string in the expected format <MODULE>:<VAR>, falling back to: {value}")
             return value
         try:
+            sys.path.insert(0, "")
             m = importlib.import_module(parts[0])
             return m.__getattribute__(parts[1])
         except ModuleNotFoundError as e: