[CI] Fix missing files in run_suite.py (#2288)

This commit is contained in:
Lianmin Zheng
2024-11-30 23:53:34 -08:00
committed by GitHub
parent 00181098dd
commit 0303ca918f

View File

@@ -15,6 +15,7 @@ suites = {
"test_double_sparsity.py", "test_double_sparsity.py",
"test_embedding_openai_server.py", "test_embedding_openai_server.py",
"test_eval_accuracy_mini.py", "test_eval_accuracy_mini.py",
"test_get_weights_by_name.py",
"test_gguf.py", "test_gguf.py",
"test_input_embeddings.py", "test_input_embeddings.py",
"test_json_constrained.py", "test_json_constrained.py",
@@ -36,10 +37,9 @@ suites = {
"test_torchao.py", "test_torchao.py",
"test_triton_attention_kernels.py", "test_triton_attention_kernels.py",
"test_triton_attention_backend.py", "test_triton_attention_backend.py",
"test_update_weights.py", "test_update_weights_from_disk.py",
"test_vision_openai_server.py", "test_vision_openai_server.py",
"test_session_control.py", "test_session_control.py",
"test_get_parameter_by_name.py",
], ],
"sampling/penaltylib": glob.glob( "sampling/penaltylib": glob.glob(
"sampling/penaltylib/**/test_*.py", recursive=True "sampling/penaltylib/**/test_*.py", recursive=True