From f4227bdb29ee00366ef566677ffbb114df8433c3 Mon Sep 17 00:00:00 2001 From: Benjamin Striner Date: Mon, 23 May 2022 13:52:14 -0400 Subject: [PATCH] Shlex quote feature: shlex quote asyncio run --- src/sagemaker_training/process.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/sagemaker_training/process.py b/src/sagemaker_training/process.py index 3a2edad0..a2aea0e7 100644 --- a/src/sagemaker_training/process.py +++ b/src/sagemaker_training/process.py @@ -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 ) @@ -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()]