EngineX-Hygon/sglang
5
0
Fork 0
Code Issues Pull Requests Actions 7 Projects Releases Wiki Activity

Fix log parsing in the chunked prefill unit tests (#1793)

Browse Source
This commit is contained in:
Lianmin Zheng
2024-10-25 08:00:55 -07:00
committed by GitHub
parent def55bc876
commit 2148914e1b
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
python/sglang/test/test_utils.py
View File

@@ -640,15 +640,17 @@ STDERR_FILENAME = "stderr.txt"
def read_output(output_lines): def read_output(output_lines):
"""Print the output in real time with another thread."""
pt = 0 pt = 0
while pt >= 0: while pt >= 0:
if pt > 0 and os.path.exists(STDERR_FILENAME): if pt > 0 and not os.path.exists(STDERR_FILENAME):
break break
lines = open(STDERR_FILENAME).readlines() lines = open(STDERR_FILENAME).readlines()
output_lines[:] = lines output_lines[:] = lines
for line in lines[pt:]: for line in lines[pt:]:
print(line, end="", flush=True) print(line, end="", flush=True)
pt += 1 pt += 1
time.sleep(0.1)
def run_mmlu_test( def run_mmlu_test(