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

update msc cli

parent 6dd99ce1
...@@ -14,10 +14,11 @@ parser = argparse.ArgumentParser( ...@@ -14,10 +14,11 @@ parser = argparse.ArgumentParser(
parser.add_argument("--dataset", type=str, help="Dataset name") 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("--project-id", type=str, help="Project ID", default="None") parser.add_argument("--project-id", type=str, help="Project ID", default="None")
parser.add_argument("--obs-type", type=str, help="Observation type", default="WIDE") parser.add_argument("--obs-type", type=str, help="Observation type", default="")
parser.add_argument("--batch-id", type=str, help="Batch ID", default="default") parser.add_argument("--batch-id", type=str, help="Batch ID", default="default")
args = parser.parse_args() args = parser.parse_args()
print(args) print("CLI parameters: ", args)
DAG_LOOP_MAP = { DAG_LOOP_MAP = {
"WIDE": ["csst-msc-l1-mbi", "csst-msc-l1-sls"], "WIDE": ["csst-msc-l1-mbi", "csst-msc-l1-sls"],
...@@ -26,6 +27,9 @@ DAG_LOOP_MAP = { ...@@ -26,6 +27,9 @@ DAG_LOOP_MAP = {
"DARK": ["csst-msc-l1-qc0"], "DARK": ["csst-msc-l1-qc0"],
"FLAT": ["csst-msc-l1-qc0"], "FLAT": ["csst-msc-l1-qc0"],
} }
if args.obs_type:
assert args.obs_type in DAG_LOOP_MAP.keys(), f"Unknown obs_type: {args.obs_type}"
DAG_LOOP_MAP = {args.obs_type: DAG_LOOP_MAP[args.obs_type]}
for obs_type, dag_ids in DAG_LOOP_MAP.items(): for obs_type, dag_ids in DAG_LOOP_MAP.items():
print(f"* Processing {obs_type}") print(f"* Processing {obs_type}")
......
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