test

Signed-off-by: Kevin Su <[email protected]>
flyteorg · Dec 3, 2024 · 935be5c · 935be5c
1 parent f07f01b
commit 935be5c
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 3 deletions.
diff --git a/flytekit/core/base_task.py b/flytekit/core/base_task.py
@@ -751,8 +751,6 @@ def dispatch_execute(
             logger.info(f"Invoking {self.name} with inputs: {native_inputs}")
             with timeit("Execute user level code"):
                 try:
-                    if hasattr(self, "run"):
-                        self.run()  # it will start vscode server
                     native_outputs = self.execute(**native_inputs)
                 except Exception as e:
                     if is_local_execution:

diff --git a/flytekit/interactive/vscode_lib/decorator.py b/flytekit/interactive/vscode_lib/decorator.py
@@ -411,7 +411,7 @@ def __init__(
             config=config,
         )
 
-    def run(self, *args, **kwargs):
+    def execute(self, *args, **kwargs):
         ctx = FlyteContextManager.current_context()
         logger = flytekit.current_context().logging
         ctx.user_space_params.builder().add_attr(