Skip to content

Commit e32cd6e

Browse files
committed
add option to call end_of_run_workflow with api_key
1 parent 84b9161 commit e32cd6e

1 file changed

Lines changed: 3 additions & 2 deletions

File tree

end_of_run_workflow.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -25,10 +25,11 @@ def log_completion(dry_run=False):
2525

2626

2727
@flow
28-
def end_of_run_workflow(stop_doc, dry_run=False):
28+
def end_of_run_workflow(stop_doc, api_key=None, dry_run=False):
2929
uid = stop_doc["run_start"]
3030
# hello_world()
31-
api_key = get_api_key_from_env(api_key=None)
31+
if not api_key:
32+
api_key = get_api_key_from_env(api_key=None)
3233
data_validation(uid, return_state=True, api_key=api_key, dry_run=dry_run)
3334
get_other_docs(uid, api_key=api_key, dry_run=dry_run)
3435
# long_flow(iterations=100, sleep_length=10, dry_run=dry_run)

0 commit comments

Comments
 (0)