Loading generate_test.py +7 −2 Original line number Diff line number Diff line Loading @@ -31,11 +31,11 @@ # import argparse from multiprocessing import Pool from pathlib import Path from ivas_processing_scripts import config from ivas_processing_scripts import main as generate_test from ivas_processing_scripts.utils import apply_func_parallel HERE = Path(__file__).parent.absolute() EXPERIMENTS_P800 = [f"P800-{i}" for i in range(1, 10)] Loading Loading @@ -64,7 +64,12 @@ def generate_tests(exp_lab_pairs, run_parallel=True, create_cfg_only=False): return args = [Arguments(str(cfg)) for cfg in cfgs] apply_func_parallel(generate_test, zip(args), type="mp" if run_parallel else None) if run_parallel: with Pool() as p: p.starmap(generate_test, zip(args), chunksize=8) else: map(generate_test, args) # apply_func_parallel(generate_test, zip(args), type="mp" if run_parallel else None) class Arguments: Loading Loading
generate_test.py +7 −2 Original line number Diff line number Diff line Loading @@ -31,11 +31,11 @@ # import argparse from multiprocessing import Pool from pathlib import Path from ivas_processing_scripts import config from ivas_processing_scripts import main as generate_test from ivas_processing_scripts.utils import apply_func_parallel HERE = Path(__file__).parent.absolute() EXPERIMENTS_P800 = [f"P800-{i}" for i in range(1, 10)] Loading Loading @@ -64,7 +64,12 @@ def generate_tests(exp_lab_pairs, run_parallel=True, create_cfg_only=False): return args = [Arguments(str(cfg)) for cfg in cfgs] apply_func_parallel(generate_test, zip(args), type="mp" if run_parallel else None) if run_parallel: with Pool() as p: p.starmap(generate_test, zip(args), chunksize=8) else: map(generate_test, args) # apply_func_parallel(generate_test, zip(args), type="mp" if run_parallel else None) class Arguments: Loading