Hi,
Thanks for your question. Right now there is no way of doing this within the workflow.

So the way we are doing this now, is passing through a task value in the workflow and checking the task value at the beginning of the notebook and use the dbutils.notebook.exit to