Skip to content

Commit fcfea61

Browse files
authored
Merge pull request #338 from xylar/fix-suite
Fix calls to write_job_script()
2 parents e143753 + 4146792 commit fcfea61

File tree

1 file changed

+15
-7
lines changed

1 file changed

+15
-7
lines changed

polaris/setup.py

Lines changed: 15 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -168,11 +168,11 @@ def setup_tasks(
168168

169169
if machine is not None:
170170
write_job_script(
171-
basic_config,
172-
machine,
173-
max_cores,
174-
max_of_min_cores,
175-
work_dir,
171+
config=basic_config,
172+
machine=machine,
173+
target_cores=max_cores,
174+
min_cores=max_of_min_cores,
175+
work_dir=work_dir,
176176
suite=suite_name,
177177
)
178178

@@ -251,7 +251,11 @@ def setup_task(
251251
cores = step.cpus_per_task * step.ntasks
252252
min_cores = step.min_cpus_per_task * step.min_tasks
253253
write_job_script(
254-
step.config, machine, cores, min_cores, step.work_dir
254+
config=step.config,
255+
machine=machine,
256+
target_cores=cores,
257+
min_cores=min_cores,
258+
work_dir=step.work_dir,
255259
)
256260
step.setup_complete = True
257261

@@ -270,7 +274,11 @@ def setup_task(
270274
if machine is not None:
271275
max_cores, max_of_min_cores = _get_required_cores({path: task})
272276
write_job_script(
273-
task.config, machine, max_cores, max_of_min_cores, task_dir
277+
config=task.config,
278+
machine=machine,
279+
target_cores=cores,
280+
min_cores=min_cores,
281+
work_dir=task_dir,
274282
)
275283

276284

0 commit comments

Comments
 (0)