Loading create_items_p800.py→experiments/create_items_p800.py +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: Loading @@ -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") Loading @@ -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) Loading
create_items_p800.py→experiments/create_items_p800.py +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: Loading @@ -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") Loading @@ -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)