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

remove plan query

parent debc2fb8
......@@ -59,8 +59,8 @@ parser = argparse.ArgumentParser(
# data level
parser.add_argument("--level", type=int, help="Data level", default=0)
# level0 data parameters
parser.add_argument("--dataset", type=str, help="Dataset name")
parser.add_argument("--instrument", type=str, help="Instrument name", default=None)
parser.add_argument("--dataset", type=str, help="Dataset name", default="csst-msc-c9-25sqdeg-v3")
parser.add_argument("--instrument", type=str, help="Instrument name", default="MSC")
parser.add_argument("--obs-type", type=str, help="Observation type", default=None)
parser.add_argument("--obs-group", type=str, help="Observation group", default=None)
parser.add_argument("--obs-id", type=str, help="Observation ID", default=None)
......@@ -91,7 +91,7 @@ print("CLI parameters: ", args)
if args.level == 0:
print("Inspecting level0 data...")
plan_basis, data_basis = Dispatcher.find_plan_level0_basis(
plan_basis, data_basis = Dispatcher.find_level0_basis(
dataset=args.dataset,
instrument=args.instrument,
obs_type=args.obs_type,
......@@ -102,7 +102,7 @@ if args.level == 0:
)
else:
print("Inspecting level1 data...")
plan_basis, data_basis = Dispatcher.find_plan_level1_basis(
plan_basis, data_basis = Dispatcher.find_level1_basis(
dataset=args.dataset,
instrument=args.instrument,
obs_type=args.obs_type,
......@@ -116,11 +116,11 @@ print(f"{len(plan_basis)} plan basis, {len(data_basis)} data basis found")
# data_basis.remove_columns(["file_name", "_id"])
stats_config = dict(
prc_status=["dataset", "prc_status"],
obs_type=["dataset", "instrument", "obs_type", "prc_status"],
obs_group=["dataset", "instrument", "obs_type", "obs_group", "prc_status"],
detector=["dataset", "instrument", "obs_type", "obs_group", "detector", "prc_status"],
obs_id=["dataset", "instrument", "obs_type", "obs_group", "obs_id", "prc_status"],
prc_status=["dataset", "prc_status", "qc_status"],
obs_type=["dataset", "instrument", "obs_type", "prc_status", "qc_status"],
obs_group=["dataset", "instrument", "obs_type", "obs_group", "prc_status", "qc_status"],
detector=["dataset", "instrument", "obs_type", "obs_group", "detector", "prc_status", "qc_status"],
obs_id=["dataset", "instrument", "obs_type", "obs_group", "obs_id", "prc_status", "qc_status"],
)
if args.level == 1:
for k in stats_config.keys():
......
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