Skip to content

Commit f8f20ed

Browse files
committed
FIXED GA: analyze_ga script wrong function calling + removed unused import.
1 parent 59ac20f commit f8f20ed

1 file changed

Lines changed: 2 additions & 3 deletions

File tree

scripts/analyze_ga_parallel_run.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,6 @@
1313
# Import codefiles
1414
from run_experiment import run_experiment, EVALUATIONS_PER_RUN
1515
from analyze_experiment import main
16-
import argparse
1716

1817
def analyze_experiment(experiment_name) -> None:
1918
main(experiment=experiment_name, diagnose_pso=False)
@@ -83,7 +82,7 @@ def allocate_job(memory_per_job_gb):
8382
run_jobs.append((model, optimizer))
8483

8584
with multiprocessing.Pool(processes=allocate_job(1)) as pool:
86-
results = pool.map(run_experiment, run_jobs)
85+
results = pool.starmap(ga_comp_run_experiment, run_jobs)
8786

8887
if not all(results):
8988
print("Some experiments failed.")
@@ -100,7 +99,7 @@ def allocate_job(memory_per_job_gb):
10099
analyze_jobs.append((model, optimizer))
101100

102101
with multiprocessing.Pool(processes=allocate_job(1)) as pool:
103-
analyze_results = pool.map(analyze_experiment, analyze_jobs)
102+
analyze_results = pool.starmap(ga_comp_analyze_experiment, analyze_jobs)
104103

105104
if not all(analyze_results):
106105
print("Some analyses failed.")

0 commit comments

Comments
 (0)