tweaks

csst_dag/cli/msc.py

@@ -65,7 +65,7 @@ for obs_type, dags in DAG_MAP.items():
     print(f"* Processing {obs_type}")
     for this_dag in dags:
         print(f"   - Scheduling `{this_dag}` -> ", end="")
-        dag = CsstDAGs.get_dag(dag=this_dag)
+        dag = CsstDAGs.get(this_dag)
         msgs = dag.schedule(
             batch_id=args.batch_id,
             priority=args.priority,

csst_dag/dag/l1.py

@@ -90,12 +90,6 @@ DAG_PARAMS = {
     },
 }
 
-SCHEDULE_KWARGS = {
-    "priority",
-    # "queue",
-    # "execution_date",
-}
-
 
 class GeneralL1DAG(BaseDAG):
 
@@ -107,15 +101,14 @@ class GeneralL1DAG(BaseDAG):
     def schedule(
         self,
         batch_id: str | None = "-",
+        priority: int = 1,
         dataset: str = "csst-msc-c9-25sqdeg-v3",
         obs_type: str = "WIDE",
         obs_group="W1",
         initial_prc_status: int = -1024,  # level0 prc_status level1
         final_prc_status: int = -2,
         demo=True,
-        **kwargs,
     ):
-        assert kwargs.keys() <= SCHEDULE_KWARGS, f"Unknown kwargs: {kwargs.keys()}"
         # no need to query plan
         # plan.write_file(local_path="plan.json")
         # plan.find(
@@ -129,7 +122,7 @@ class GeneralL1DAG(BaseDAG):
         dag_group_run = self.gen_dag_group_run(
             dag_group=self.dag_group,
             batch_id=batch_id,
-            priority=kwargs.get("priority", 1),
+            priority=priority,
         )
         if demo:
             print(json.dumps(dag_group_run, indent=4))