From 0303ca918fdf7b87df63c8902a36f2623cb4dea0 Mon Sep 17 00:00:00 2001
From: Lianmin Zheng <lianminzheng@gmail.com>
Date: Sat, 30 Nov 2024 23:53:34 -0800
Subject: [PATCH] [CI] Fix missing files in run_suite.py (#2288)

---
 test/srt/run_suite.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/test/srt/run_suite.py b/test/srt/run_suite.py
index b055df1a1..18064ca27 100644
--- a/test/srt/run_suite.py
+++ b/test/srt/run_suite.py
@@ -15,6 +15,7 @@ suites = {
         "test_double_sparsity.py",
         "test_embedding_openai_server.py",
         "test_eval_accuracy_mini.py",
+        "test_get_weights_by_name.py",
         "test_gguf.py",
         "test_input_embeddings.py",
         "test_json_constrained.py",
@@ -36,10 +37,9 @@ suites = {
         "test_torchao.py",
         "test_triton_attention_kernels.py",
         "test_triton_attention_backend.py",
-        "test_update_weights.py",
+        "test_update_weights_from_disk.py",
         "test_vision_openai_server.py",
         "test_session_control.py",
-        "test_get_parameter_by_name.py",
     ],
     "sampling/penaltylib": glob.glob(
         "sampling/penaltylib/**/test_*.py", recursive=True