Skip to content

Commit

Permalink
Shlex quote
Browse files Browse the repository at this point in the history
feature: shlex quote asyncio run
  • Loading branch information
bstriner committed May 23, 2022
1 parent 22a170a commit 4e87e01
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions src/sagemaker_training/process.py
Original file line number Diff line number Diff line change
Expand Up @@ -113,7 +113,7 @@ async def run_async(cmd, processes_per_host, env, cwd, stderr, **kwargs):
Raises:
error_class: If there is an exception raised when creating the process.
"""
cmd = " ".join(cmd)
cmd = " ".join(six.moves.shlex_quote(token) for token in cmd)
proc = await asyncio.create_subprocess_shell(
cmd, env=env, cwd=cwd, stdout=PIPE, stderr=stderr, **kwargs
)
Expand Down Expand Up @@ -255,7 +255,7 @@ def _create_command(self):
six.moves.shlex_quote(arg) # pylint: disable=too-many-function-args
for arg in self._args
]
return ["/bin/sh", "-c", '"./%s %s"' % (self._user_entry_point, " ".join(args))]
return ["/bin/sh", "-c", './%s %s' % (self._user_entry_point, " ".join(args))]

def _python_command(self): # pylint: disable=no-self-use
return [python_executable()]
Expand Down

0 comments on commit 4e87e01

Please sign in to comment.