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):