From 70996373037e901b69f34e9fe25f62977d8249e4 Mon Sep 17 00:00:00 2001 From: Calvin Leather Date: Fri, 7 Oct 2022 09:19:36 -0400 Subject: [PATCH] Add notebook to default deck rather than new deck --- plugins/flytekit-papermill/flytekitplugins/papermill/task.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/plugins/flytekit-papermill/flytekitplugins/papermill/task.py b/plugins/flytekit-papermill/flytekitplugins/papermill/task.py index 0721c39a37..d9484721d2 100644 --- a/plugins/flytekit-papermill/flytekitplugins/papermill/task.py +++ b/plugins/flytekit-papermill/flytekitplugins/papermill/task.py @@ -241,7 +241,8 @@ def execute(self, **kwargs) -> Any: output_list = [] for k, type_v in self.python_interface.outputs.items(): if k == self._IMPLICIT_OP_NOTEBOOK: - output_list.append(self.output_notebook_path) + output_list + (self.output_notebook_path) elif k == self._IMPLICIT_RENDERED_NOTEBOOK: output_list.append(self.rendered_output_path) elif k in m: @@ -260,7 +261,7 @@ def post_execute(self, user_params: ExecutionParameters, rval: Any) -> Any: nb_deck.append(notebook_html) # Since user_params is passed by reference, this modifies the object in the outside scope # which then causes the deck to be rendered later during the dispatch_execute function. - user_params.decks.append(nb_deck) + user_params.default_deck.append(nb_deck) return self._config_task_instance.post_execute(user_params, rval)