Commit 89da0f2a authored by Jan Kiene's avatar Jan Kiene
Browse files

move p800 category-wise processing wrapper script

parent 622b8f42
Loading
Loading
Loading
Loading
+7 −6
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
from ivas_processing_scripts.utils import apply_func_parallel

P800_TESTS = [f"P800-{i}" for i in range(1, 8)]
LABS = ["a", "b", "c", "d"]


class Arguments:
@@ -16,9 +16,9 @@ class Arguments:
        self.multiprocessing = False


def create_items(testname):
    p800_path = Path(f"experiments/selection/{testname}")
    p800_cfgs = p800_path.joinpath("config").glob("P800*cat*.yml")
def create_items(experiment, lab):
    p800_path = Path(f"experiments/selection/{experiment}")
    p800_cfgs = p800_path.joinpath("config").glob(f"{experiment}cat*-lab_{lab}.yml")

    args = [Arguments(str(cfg)) for cfg in p800_cfgs]
    apply_func_parallel(generate_test, zip(args), type="mp")
@@ -27,7 +27,8 @@ def create_items(testname):
# if is necessary here so that multiprocessing does not crash
if __name__ == "__main__":
    parser = argparse.ArgumentParser()
    parser.add_argument("testname", choices=P800_TESTS)
    parser.add_argument("experiment", choices=P800_TESTS)
    parser.add_argument("lab", choices=LABS)
    args = parser.parse_args()

    create_items(args.testname)
    create_items(args.experiment, args.lab)