diff --git a/test/srt/test_torchao.py b/test/srt/test_torchao.py index 765567136..2a2fcb8df 100644 --- a/test/srt/test_torchao.py +++ b/test/srt/test_torchao.py @@ -13,7 +13,7 @@ from sglang.test.test_utils import ( ) -class TestTorchCompile(unittest.TestCase): +class TestTorchAO(unittest.TestCase): @classmethod def setUpClass(cls): cls.model = DEFAULT_MODEL_NAME_FOR_TEST