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

2024-10-25 08:00:55 -07:00
parent def55bc876
commit 2148914e1b
1 changed files with 3 additions and 1 deletions
--- a/python/sglang/test/test_utils.py
+++ b/python/sglang/test/test_utils.py
@@ -640,15 +640,17 @@ STDERR_FILENAME = "stderr.txt"


 def read_output(output_lines):
+    """Print the output in real time with another thread."""
    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
        lines = open(STDERR_FILENAME).readlines()
        output_lines[:] = lines
        for line in lines[pt:]:
            print(line, end="", flush=True)
            pt += 1
+        time.sleep(0.1)


 def run_mmlu_test(