ClearML/clearml-agent
1
0
Fork 0
mirror of https://github.com/clearml/clearml-agent synced 2025-06-03 11:16:19 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Support raise_error in get_bash_output()

Browse Source
This commit is contained in:
allegroai 2022-12-05 11:26:40 +02:00
parent 57cde21c48
commit 53d379205f
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
clearml_agent/helper/process.py
View File

@ -25,7 +25,7 @@ from clearml_agent.helper.base import bash_c, is_windows_platform, select_for_pl
PathLike = Union[Text, Path] PathLike = Union[Text, Path]
def get_bash_output(cmd, strip=False, stderr=subprocess.STDOUT, stdin=False): def get_bash_output(cmd, strip=False, stderr=subprocess.STDOUT, stdin=False, raise_error=False):
try: try:
output = ( output = (
subprocess.check_output( subprocess.check_output(
@ -37,6 +37,8 @@ def get_bash_output(cmd, strip=False, stderr=subprocess.STDOUT, stdin=False):
.strip() .strip()
) )
except subprocess.CalledProcessError: except subprocess.CalledProcessError:
if raise_error:
raise
output = None output = None
return output if not strip or not output else output.strip() return output if not strip or not output else output.strip()