Adapt fixture-kit to gsm8k mixin (#15599)

This commit is contained in:
Liangsheng Yin
2025-12-22 14:19:35 +08:00
committed by GitHub
parent 1167867e8d
commit beae3f961c
8 changed files with 114 additions and 66 deletions
+3 -3
View File
@@ -5,8 +5,8 @@ import unittest
from types import SimpleNamespace
from sglang.srt.utils import is_hip
from sglang.test.mmmu_vlm_mixin import DEFAULT_MEM_FRACTION_STATIC, MMMUVLMMixin
from sglang.test.test_utils import CustomTestCase, is_in_ci
from sglang.test.kits.mmmu_vlm_kit import DEFAULT_MEM_FRACTION_STATIC, MMMUVLMTestBase
from sglang.test.test_utils import is_in_ci
_is_hip = is_hip()
# VLM models for testing
@@ -20,7 +20,7 @@ else:
]
class TestVLMModels(MMMUVLMMixin, CustomTestCase):
class TestVLMModels(MMMUVLMTestBase):
def test_vlm_mmmu_benchmark(self):
"""Test VLM models against MMMU benchmark."""
models_to_test = MODELS