Commit 2548b402 authored by Archit Tamarapu's avatar Archit Tamarapu
Browse files

add change to generate_test.py too

parent 6124103f
Loading
Loading
Loading
Loading
+7 −2
Original line number Diff line number Diff line
@@ -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)]
@@ -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: