From aae5434bdffd13dc9e00417379168146602553e5 Mon Sep 17 00:00:00 2001 From: Lianmin Zheng Date: Thu, 14 Nov 2024 11:08:37 -0800 Subject: [PATCH] Fix unit tests (#2034) --- python/sglang/test/test_utils.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/python/sglang/test/test_utils.py b/python/sglang/test/test_utils.py index 364ccefc5..ec85a12c3 100644 --- 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