From d29f2cd8670c455f6d0ec70c674381ecc3e51910 Mon Sep 17 00:00:00 2001 From: Jun Aishima Date: Tue, 31 Mar 2026 14:43:04 -0400 Subject: [PATCH] pass arguments to decorated function --- end_of_run_workflow.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/end_of_run_workflow.py b/end_of_run_workflow.py index 8b5cae0..e709cc8 100644 --- a/end_of_run_workflow.py +++ b/end_of_run_workflow.py @@ -31,7 +31,7 @@ def slack(func): the flow. To keep the naming of workflows consistent, the name of this inner function had to match the expected name. """ - def end_of_run_workflow(stop_doc, api_key=None): + def end_of_run_workflow(stop_doc, api_key=None, dry_run=False): flow_run_name = FlowRunContext.get().flow_run.dict().get("name") # Load slack credentials that are saved in Prefect. @@ -52,7 +52,7 @@ def end_of_run_workflow(stop_doc, api_key=None): ) try: - result = func(stop_doc) + result = func(stop_doc, api_key=api_key, dry_run=dry_run) # Send a message to mon-prefect if flow-run is successful. mon_prefect.notify(