Fix unit tests (#2034)

2024-11-14 11:08:37 -08:00
parent c3eac1b010
commit aae5434bdf
1 changed files with 1 additions and 2 deletions
--- 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