Commit 3f5bef29 authored by BO ZHANG's avatar BO ZHANG 🏀
Browse files

tweaks

parent b2abedfc

csst_dag/cli/msc.py

+3 −3
Original line number Diff line number Diff line
@@ -64,9 +64,9 @@ if args.obs_type: 
for obs_type, dags in DAG_MAP.items():

    print(f"* Processing `{obs_type}`")

    for this_dag in dags:

        print(f"  - Scheduling `{this_dag}` -> ", end="")

        print(f"  - Scheduling `{this_dag}`")

        dag = CsstDAGs().get(this_dag)

        msgs = dag.schedule(

        dag_group_run, dag_run_list = dag.schedule(

            batch_id=args.batch_id,

            priority=args.priority,

            dataset=args.dataset,
@@ -76,4 +76,4 @@ for obs_type, dags in DAG_MAP.items(): 
            final_prc_status=args.final_prc_status,

            demo=args.demo,

        )

        print(f"{len(msgs)} tasks.")
 
        # print(f"{len(dag_run_list)} tasks.")

csst_dag/dag/l1.py

+4 −1
Original line number Diff line number Diff line
@@ -179,6 +179,9 @@ class GeneralL1DAG(BaseDAG): 
        if not demo:

            # push and update

            res_push = self.push_dag_group_run(dag_group_run, dag_run_list)

            print(res_push)

            print(f"    * {len(dag_run_list)} DAG runs -> {res_push}")

            assert res_push.success, res_push.message

        else:

            # no push

            print(f"    * {len(dag_run_list)} DAG runs.")

        return dag_group_run, dag_run_list