Crash the server on warnings in CI (#1772)

This commit is contained in:
Lianmin Zheng
2024-10-23 16:27:13 -07:00
committed by GitHub
parent 3f5ac88d02
commit 05b3bf5e8e
5 changed files with 22 additions and 6 deletions

View File

@@ -1,3 +1,8 @@
"""
Usage:
python -m unittest test_eval_accuracy_large.TestEvalAccuracyLarge.test_mmlu
"""
import unittest
from types import SimpleNamespace
@@ -32,12 +37,12 @@ class TestEvalAccuracyLarge(unittest.TestCase):
base_url=self.base_url,
model=self.model,
eval_name="mmlu",
num_examples=3000,
num_examples=5000,
num_threads=1024,
)
metrics = run_eval(args)
assert metrics["score"] >= 0.705, f"{metrics}"
assert metrics["score"] >= 0.71, f"{metrics}"
def test_human_eval(self):
args = SimpleNamespace(

View File

@@ -1,3 +1,8 @@
"""
Usage:
python -m unittest test_moe_eval_accuracy_large.TestMoEEvalAccuracyLarge.test_mmlu
"""
import unittest
from types import SimpleNamespace
@@ -11,7 +16,7 @@ from sglang.test.test_utils import (
)
class TestEvalAccuracyLarge(unittest.TestCase):
class TestMoEEvalAccuracyLarge(unittest.TestCase):
@classmethod
def setUpClass(cls):
cls.model = DEFAULT_MOE_MODEL_NAME_FOR_TEST
@@ -37,7 +42,7 @@ class TestEvalAccuracyLarge(unittest.TestCase):
base_url=self.base_url,
model=self.model,
eval_name="mmlu",
num_examples=3000,
num_examples=5000,
num_threads=1024,
)

View File

@@ -1,4 +1,3 @@
import json
import unittest
from types import SimpleNamespace