[CI] Minor fix for CI (#2187)

2024-11-25 16:38:43 -08:00
parent 1aea19f64b
commit 1605ae121e
3 changed files with 6 additions and 5 deletions
--- a/test/srt/run_suite.py
+++ b/test/srt/run_suite.py
@@ -25,6 +25,7 @@ suites = {
        "test_radix_attention.py",
        "test_retract_decode.py",
        "test_server_args.py",
+        "test_session_control.py",
        "test_skip_tokenizer_init.py",
        "test_srt_engine.py",
        "test_srt_endpoint.py",
@@ -35,7 +36,6 @@ suites = {
        "test_triton_attention_backend.py",
        "test_update_weights.py",
        "test_vision_openai_server.py",
-        "test_session_control.py",
    ],
    "sampling/penaltylib": glob.glob(
        "sampling/penaltylib/**/test_*.py", recursive=True