reset.py 719 Bytes
Newer Older
BO ZHANG's avatar
BO ZHANG committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
"""
Aim
---
Reset a dataset.

Example
-------
python -m csst_dag.cli.reset -h

python -m csst_dag.cli.reset \
    --dataset=csst-msc-c9-25sqdeg-v3 \
    --final-prc-status=-2
"""

from csst_dag.dag import CsstDAGs
from csst_dfs_client import level0
import argparse
import tqdm

parser = argparse.ArgumentParser(
    description="Scheduler for MSC L1 pipeline.",
    formatter_class=argparse.ArgumentDefaultsHelpFormatter,
)

# data related parameters
parser.add_argument("--dataset", type=str, help="Dataset name")
parser.add_argument(
    "--final-prc-status", type=int, help="Final processing status", default=-2
)

args = parser.parse_args()
print("CLI parameters: ", args)

recs = level0.find(dataset=args.dataset)