Commit e8e7ad6b authored by Jan Kiene's avatar Jan Kiene
Browse files

add wrapper script for generating P800 items category-wise

parent bd2b65a1
Loading
Loading
Loading
Loading

create_items_p800.py

0 → 100644
+26 −0
Original line number Diff line number Diff line
#! /usr/bin/env python3
import argparse
from pathlib import Path
from ivas_processing_scripts import main as generate_test

P800_TESTS = [f"P800-{i}" for i in range(1, 8)]


class Arguments:
    def __init__(self, config):
        self.config = config
        self.debug = False


# if is necessary here so that multiprocessing does not crash
if __name__ == "__main__":
    parser = argparse.ArgumentParser()
    parser.add_argument("testname", choices=P800_TESTS)
    args = parser.parse_args()

    p800_path = Path(f"experiments/selection/{args.testname}")
    p800_cfgs = p800_path.joinpath("config").glob("P800*cat*.yml")

    for cfg in p800_cfgs:
        args_dummy = Arguments(str(cfg))
        generate_test(args_dummy)
 No newline at end of file