Fix unit tests (#2034)

aae5434b · Lianmin Zheng · GitHub · c3eac1b0 · aae5434b
Unverified Commit aae5434b authored Nov 14, 2024 by Lianmin Zheng Committed by GitHub Nov 14, 2024
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 2 deletions

python/sglang/test/test_utils.py python/sglang/test/test_utils.py +1 -2

No files found.
--- a/python/sglang/test/test_utils.py
+++ b/python/sglang/test/test_utils.py
@@ -740,8 +740,7 @@ def run_mmlu_test(

        try:
            metrics = run_eval(args)
-            print(f"{metrics=}")
-            self.assertGreaterEqual(metrics["score"], 0.65)
+            assert metrics["score"] >= 0.65, f"{metrics=}"
        finally:
            pass