From 2239968e3f82e15957ef7d2dd4c29885d29321b1 Mon Sep 17 00:00:00 2001 From: Yicheng-Lu-llll Date: Thu, 22 Jun 2023 00:28:21 +0000 Subject: [PATCH] lint Signed-off-by: Yicheng-Lu-llll --- flytekit/bin/entrypoint.py | 1 + flytekit/deck/deck.py | 4 +--- 2 files changed, 2 insertions(+), 3 deletions(-) diff --git a/flytekit/bin/entrypoint.py b/flytekit/bin/entrypoint.py index c47313b805..a2b21260ed 100644 --- a/flytekit/bin/entrypoint.py +++ b/flytekit/bin/entrypoint.py @@ -159,6 +159,7 @@ def _dispatch_execute( ctx.file_access.put_data(ctx.execution_state.engine_dir, output_prefix, is_multipart=True) logger.info(f"Engine folder written successfully to the output prefix {output_prefix}") + if not task_def.disable_deck: _output_deck(task_def.name.split(".")[-1], ctx.user_space_params) diff --git a/flytekit/deck/deck.py b/flytekit/deck/deck.py index c1c7e990ee..024fff2479 100644 --- a/flytekit/deck/deck.py +++ b/flytekit/deck/deck.py @@ -76,9 +76,7 @@ def html(self) -> str: def persist(self): task_name = FlyteContextManager.current_context().user_space_params.task_id.name new_user_params = FlyteContextManager.current_context().user_space_params - _output_deck(task_name , new_user_params) - - + _output_deck(task_name, new_user_params) class TimeLineDeck(Deck):