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

add batch_id and build for level1 inspect

parent fc117eed
Loading
Loading
Loading
Loading
+12 −7
Original line number Diff line number Diff line
@@ -115,13 +115,18 @@ else:
print(f"{len(plan_basis)} plan basis, {len(data_basis)} data basis found")
# data_basis.remove_columns(["file_name", "_id"])

for stats_type, stats_keys in dict(
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"],
).items():
)
if args.level == 1:
    for k in stats_config.keys():
        stats_config[k].extend(["batch_id", "build"])

for stats_type, stats_keys in stats_config.items():
    if args.verbose:
        data_basis.pprint_all()
    print("\n>>> STATS DIM: ", stats_type)