Skip to content

Commit 8293b8a

Browse files
committed
fix: getting resource args from map
1 parent 2bfc45d commit 8293b8a

1 file changed

Lines changed: 4 additions & 5 deletions

File tree

wfcommons/wfbench/translator/nextflow.py

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -101,7 +101,7 @@ def translate(self, output_file_path: pathlib.Path) -> None:
101101

102102
def _is_resource_arg(self, arg: str) -> bool:
103103
return arg.startswith("--percent-cpu") or arg.startswith("--mem") \
104-
or arg.startswith("--cpu_work") or arg.startswith("--gpu_work")
104+
or arg.startswith("--cpu-work") or arg.startswith("--gpu-work")
105105

106106
def _determine_abstract_relations(self) -> None:
107107
"""
@@ -206,19 +206,18 @@ def _add_abstract_task_definition(self, abstract_task_name: str, physical_tasks:
206206
if a == f"{abstract_task_name}_{example_task.task_id}":
207207
cmd += f" {abstract_task_name}_${{id}}"
208208
continue
209-
cmd += ' '
210209
if self._is_resource_arg(a):
211210
if resource_args_done:
212211
continue
213-
cmd += self.valid_task_name(abstract_task_name) + "_args.get(id).get(\"resources\") "
212+
cmd += " " + self.valid_task_name(abstract_task_name) + "_args.get(id).get(\"resources\")"
214213
resource_args_done = True
215214
elif a.startswith("--out"):
216-
cmd += "--out \"{${" + self.valid_task_name(abstract_task_name) + "_args.get(id).get(\"out\")}}\""
215+
cmd += " --out \"{${" + self.valid_task_name(abstract_task_name) + "_args.get(id).get(\"out\")}}\""
217216
cmd += " \\$inputs"
218217
break
219218
else:
220219
a = a.replace(f"{abstract_task_name}_{example_task.task_id}", f"{abstract_task_name}_${{id}}")
221-
cmd += a.replace("'", "\"")
220+
cmd += " " + a.replace("'", "\"")
222221

223222
# creating the abstract task
224223
self.script += f"process task_{self.valid_task_name(abstract_task_name)}" + " {\n"

0 commit comments

Comments
 (0)