@@ -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