tweaks

csst_dag/cli/msc.py

@@ -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

@@ -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