onnxruntime/tools/python/util/run.py
Justin Chu d64769c38e
Set black's target version (#11370)
Description: Set black's target version to be py37 - py310

Motivation and Context

Black by default targets its format for py3.10. Since our project supports python 3.7, we need to target version to all the python versions supported.

Re-ran black. 13 files reformatted.
2022-04-27 14:52:19 -07:00

62 lines
1.6 KiB
Python

# Copyright (c) Microsoft Corporation. All rights reserved.
# Licensed under the MIT License.
import logging
import os
import shlex
import subprocess
_log = logging.getLogger("util.run")
def run(
*args,
cwd=None,
input=None,
capture_stdout=False,
capture_stderr=False,
shell=False,
env=None,
check=True,
quiet=False,
):
"""Runs a subprocess.
Args:
*args: The subprocess arguments.
cwd: The working directory. If None, specifies the current directory.
input: The optional input byte sequence.
capture_stdout: Whether to capture stdout.
capture_stderr: Whether to capture stderr.
shell: Whether to run using the shell.
env: The environment variables as a dict. If None, inherits the current
environment.
check: Whether to raise an error if the return code is not zero.
quiet: If true, do not print output from the subprocess.
Returns:
A subprocess.CompletedProcess instance.
"""
cmd = [*args]
_log.info(
"Running subprocess in '{0}'\n {1}".format(cwd or os.getcwd(), " ".join([shlex.quote(arg) for arg in cmd]))
)
def output(is_stream_captured):
return subprocess.PIPE if is_stream_captured else (subprocess.DEVNULL if quiet else None)
completed_process = subprocess.run(
cmd,
cwd=cwd,
check=check,
input=input,
stdout=output(capture_stdout),
stderr=output(capture_stderr),
env=env,
shell=shell,
)
_log.debug("Subprocess completed. Return code: {}".format(completed_process.returncode))
return completed_process