tweaks

    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,

    },

}

SCHEDULE_KWARGS = {

    "priority",

    # "queue",

    # "execution_date",

}

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(

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