Fix test and benchmark scripts (#2598)
This commit is contained in:
@@ -4,7 +4,7 @@ import glob
|
||||
from sglang.test.test_utils import run_unittest_files
|
||||
|
||||
suites = {
|
||||
"minimal": ["test_srt_backend.py", "test_openai_backend.py"],
|
||||
"per-commit": ["test_srt_backend.py", "test_openai_backend.py"],
|
||||
}
|
||||
|
||||
|
||||
|
||||
@@ -4,7 +4,7 @@ import glob
|
||||
from sglang.test.test_utils import run_unittest_files
|
||||
|
||||
suites = {
|
||||
"minimal": [
|
||||
"per-commit": [
|
||||
"models/test_embedding_models.py",
|
||||
"models/test_generation_models.py",
|
||||
"models/test_lora.py",
|
||||
|
||||
@@ -30,7 +30,7 @@ class TestTritonAttnBackend(unittest.TestCase):
|
||||
)
|
||||
|
||||
if is_in_ci():
|
||||
assert output_throughput > 153, f"{output_throughput=}"
|
||||
self.assertGreater(output_throughput, 153)
|
||||
|
||||
def test_mmlu(self):
|
||||
model = DEFAULT_MODEL_NAME_FOR_TEST
|
||||
|
||||
Reference in New Issue
Block a user