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

tweaks

parent 9188a8dd
""" """
Aim Aim
--- ---
Process an MSC dataset, given Process an MSC dataset, given a set of parameters.
- project_id = "none"
- obs_type = ""
- batch_id = "csci-test-20250507"
Example Example
------- -------
python -m csst_dag.cli.msc --dataset=csst-msc-c9-25sqdeg-v3 --project-id=none --batch-id=csci-test-20250507 python -m csst_dag.cli.msc \
--dataset=csst-msc-c9-25sqdeg-v3 \
--project-id=none \
--batch-id=csci-test-20250507 \
--priority=1 \
--initial-prc-status=-1024 \
--final-prc-status=-2
""" """
from csst_dag.dag import CsstDAG from csst_dag.dag import CsstDAG
import argparse import argparse
# python -m csst_dag.trigger.csst-msc-l1-mbi -h
# python -m csst_dag.trigger.csst-msc-l1-mbi --dataset csst-msc-c9-25sqdeg-v3 --batch-id
parser = argparse.ArgumentParser( parser = argparse.ArgumentParser(
description="Scheduler for MSC L1 pipeline.", description="Scheduler for MSC L1 pipeline.",
formatter_class=argparse.ArgumentDefaultsHelpFormatter, formatter_class=argparse.ArgumentDefaultsHelpFormatter,
...@@ -25,7 +26,14 @@ parser.add_argument("--dataset", type=str, help="Dataset name") ...@@ -25,7 +26,14 @@ parser.add_argument("--dataset", type=str, help="Dataset name")
# parser.add_argument("--instrument", type=str, help="Instrument name", default="MSC") # parser.add_argument("--instrument", type=str, help="Instrument name", default="MSC")
parser.add_argument("--obs-type", type=str, help="Observation type", default="") parser.add_argument("--obs-type", type=str, help="Observation type", default="")
parser.add_argument("--project-id", type=str, help="Project ID", default="none") parser.add_argument("--project-id", type=str, help="Project ID", default="none")
parser.add_argument("--batch-id", type=str, help="Batch ID", default="default") parser.add_argument("--batch-id", type=str, help="Batch ID", default="default_batch")
parser.add_argument("--priority", type=str, help="Task priority", default=1)
parser.add_argument(
"--initial-prc-status", type=int, help="Initial processing status", default=-1024
)
parser.add_argument(
"--final-prc-status", type=int, help="Final processing status", default=-2
)
args = parser.parse_args() args = parser.parse_args()
print("CLI parameters: ", args) print("CLI parameters: ", args)
...@@ -51,8 +59,9 @@ for obs_type, dag_ids in DAG_LOOP_MAP.items(): ...@@ -51,8 +59,9 @@ for obs_type, dag_ids in DAG_LOOP_MAP.items():
obs_type=obs_type, obs_type=obs_type,
project_id=args.project_id, project_id=args.project_id,
batch_id=args.batch_id, batch_id=args.batch_id,
initial_prc_status=-2, initial_prc_status=args.initial_prc_status,
final_prc_status=-2, final_prc_status=args.final_prc_status,
demo=True, demo=True,
priority=args.priority,
) )
print(f"{len(msgs)} tasks.") print(f"{len(msgs)} tasks.")
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment