From eb6bcc5c868248b64c613b0a0d2b43d45dde39bc Mon Sep 17 00:00:00 2001 From: Alison Shao <54658187+alisonshao@users.noreply.github.com> Date: Tue, 3 Mar 2026 16:53:31 -0800 Subject: [PATCH] [CI] Register test_quant_config_parsing.py in CI suite (#19809) Co-authored-by: Alison Shao --- .../quant}/test_quant_config_parsing.py | 13 +++++-------- 1 file changed, 5 insertions(+), 8 deletions(-) rename test/{srt => registered/quant}/test_quant_config_parsing.py (91%) diff --git a/test/srt/test_quant_config_parsing.py b/test/registered/quant/test_quant_config_parsing.py similarity index 91% rename from test/srt/test_quant_config_parsing.py rename to test/registered/quant/test_quant_config_parsing.py index 40f130b46..f863a13c7 100644 --- a/test/srt/test_quant_config_parsing.py +++ b/test/registered/quant/test_quant_config_parsing.py @@ -2,16 +2,13 @@ import unittest from unittest.mock import MagicMock from sglang.srt.configs.model_config import ModelConfig +from sglang.test.ci.ci_register import register_cpu_ci +from sglang.test.test_utils import CustomTestCase + +register_cpu_ci(est_time=5, suite="stage-a-cpu-only") -class MockHfConfig: - def __init__(self, quant_config=None): - self.quantization_config = quant_config - self.architectures = ["LlamaForCausalLM"] - self.model_type = "llama" - - -class TestQuantLogString(unittest.TestCase): +class TestQuantLogString(CustomTestCase): def test_qwen_fp8_config(self): # Example from Qwen/Qwen3-4B-Thinking-2507-FP8 quant_config = {