Commit c1becf46 authored by BO ZHANG's avatar BO ZHANG 🏀
Browse files

add --dags to select DAGs

parent f403e059

csst_dag/cli/csst_msc_l1.py

+1 −1
Original line number Diff line number Diff line
@@ -134,7 +134,7 @@ DEFAULT_DAGS = { 
    "csst-msc-l1-sls",

    "csst-msc-l1-qc0",

}

SELECTED_DAGS = DEFAULT_DAGS if args.dag is None else {dag for dag in args.dags if dag in DEFAULT_DAGS}

SELECTED_DAGS = DEFAULT_DAGS if args.dags is None else {dag for dag in args.dags if dag in DEFAULT_DAGS}

print("Selected DAGs: ", SELECTED_DAGS)

for dag in SELECTED_DAGS:

    this_task_list = CSST_DAGS[dag].schedule(