From 28b4d8e14472eda2f4e779c2e3a7f9eac5a522b3 Mon Sep 17 00:00:00 2001 From: Lianmin Zheng Date: Tue, 24 Sep 2024 03:17:10 -0700 Subject: [PATCH] Update test_srt_backend.py (#1502) --- test/lang/test_srt_backend.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/test/lang/test_srt_backend.py b/test/lang/test_srt_backend.py index 62c595928..5bc565c18 100644 --- a/test/lang/test_srt_backend.py +++ b/test/lang/test_srt_backend.py @@ -67,7 +67,7 @@ class TestSRTBackend(unittest.TestCase): # Run twice to capture more bugs for _ in range(2): accuracy, latency = test_hellaswag_select() - assert accuracy > 0.71 + assert accuracy > 0.71, f"{accuracy=}" if __name__ == "__main__":