update code

Author: Jintao-Huang

tests/megatron/test_align/test_llm.py

@@ -137,7 +137,7 @@ def test_ernie():
     # test_llama3_2()
     # test_qwen3()
     # test_qwen2_moe()
-    # test_qwen3_moe()
+    test_qwen3_moe()
     # test_internlm3()
     # test_mimo()
     # test_moonlight()
@@ -146,4 +146,4 @@ def test_ernie():
     # test_dots()
     # test_kimi_dev()
     # test_hunyuan()
-    test_ernie()
+    # test_ernie()

tests/test_align/test_template/test_llm.py

@@ -446,15 +446,6 @@ def test_ernie():
     assert res == res2, f'res: {res}, res2: {res2}'
 
 
-def test_glm4_5():
-    messages = [{'role': 'user', 'content': '浙江的省会在哪?'}]
-    pt_engine = PtEngine('ZhipuAI/GLM-4.5-Air')
-    res = _infer_model(pt_engine, messages=messages)
-    pt_engine.default_template.template_backend = 'jinja'
-    res2 = _infer_model(pt_engine, messages=messages)
-    assert res == res2, f'res: {res}, res2: {res2}'
-
-
 if __name__ == '__main__':
     from swift.llm import PtEngine, RequestConfig
     from swift.utils import get_logger, seed_everything
@@ -499,4 +490,3 @@ def test_glm4_5():
     # test_kimi_dev()
     # test_hunyuan()
     # test_ernie()
-    test_glm4_5()