SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

improve error checking (#8157)

Browse Source
This commit is contained in:
Stas Bekman
2020-10-29 11:05:24 -07:00
committed by GitHub
parent 691176283d
commit c83cec44f8
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
src/transformers/testing_utils.py
View File

@@ -849,7 +849,7 @@ async def _stream_subprocess(cmd, env=None, stdin=None, timeout=None, quiet=Fals
# XXX: the timeout doesn't seem to make any difference here # XXX: the timeout doesn't seem to make any difference here
await asyncio.wait( await asyncio.wait(
[ [
_read_stream(p.stdout, lambda l: tee(l, out, sys.stdout)), _read_stream(p.stdout, lambda l: tee(l, out, sys.stdout, label="stdout:")),
_read_stream(p.stderr, lambda l: tee(l, err, sys.stderr, label="stderr:")), _read_stream(p.stderr, lambda l: tee(l, err, sys.stderr, label="stderr:")),
], ],
timeout=timeout, timeout=timeout,
@@ -869,7 +869,10 @@ def execute_subprocess_async(cmd, env=None, stdin=None, timeout=180, quiet=False
raise RuntimeError( raise RuntimeError(
f"'{cmd_str}' failed with returncode {result.returncode} - see the `stderr:` messages from above for details." f"'{cmd_str}' failed with returncode {result.returncode} - see the `stderr:` messages from above for details."
) )
if not result.stdout:
# check that the subprocess actually did run and produced some output, should the test rely on
# the remote side to do the testing
if not result.stdout and not result.stderr:
raise RuntimeError(f"'{cmd_str}' produced no output.") raise RuntimeError(f"'{cmd_str}' produced no output.")
return result return result