From b33e1ba5d9357b72eea541f8dabd700df4f38a7c Mon Sep 17 00:00:00 2001 From: Future Outlier Date: Thu, 30 Nov 2023 16:33:52 +0800 Subject: [PATCH] lint Signed-off-by: Future Outlier --- .../flytekitplugins/flyin/vscode_lib/decorator.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/plugins/flytekit-flyin/flytekitplugins/flyin/vscode_lib/decorator.py b/plugins/flytekit-flyin/flytekitplugins/flyin/vscode_lib/decorator.py index 9c461ecc31..7b2ac72cc9 100644 --- a/plugins/flytekit-flyin/flytekitplugins/flyin/vscode_lib/decorator.py +++ b/plugins/flytekit-flyin/flytekitplugins/flyin/vscode_lib/decorator.py @@ -13,15 +13,15 @@ import flytekit from flytekit.core.context_manager import FlyteContextManager -import flytekit + from .config import VscodeConfig from .constants import ( DOWNLOAD_DIR, EXECUTABLE_NAME, HEARTBEAT_CHECK_SECONDS, HEARTBEAT_PATH, - MAX_IDLE_SECONDS, INTERACTIVE_DEBUGGING_FILE_NAME, + MAX_IDLE_SECONDS, ) @@ -170,7 +170,10 @@ def prepare_interactive_python(task_function): context_working_dir = FlyteContextManager.current_context().execution_state.working_dir # Copy the user's Python file to the working directory. - shutil.copy(f"{task_function.__module__}.py", os.path.join(context_working_dir, f"{task_function.__module__}.py")) + shutil.copy( + f"{task_function.__module__}.py", + os.path.join(context_working_dir, f"{task_function.__module__}.py"), + ) # Generate a Python script task_module_name, task_name = task_function.__module__, task_function.__name__