From a2c45d82c362da748f0491a235ed8893b8d19e0e Mon Sep 17 00:00:00 2001 From: JunyiXu-nv <219237550+JunyiXu-nv@users.noreply.github.com> Date: Mon, 15 Sep 2025 12:16:44 +0800 Subject: [PATCH] [None][chore] Enable multiple postprocess workers tests for chat completions api (#7602) Signed-off-by: Junyi Xu <219237550+JunyiXu-nv@users.noreply.github.com> --- .../llmapi/apps/_test_openai_chat_harmony.py | 12 ++++++++++-- 1 file changed, 10 insertions(+), 2 deletions(-) diff --git a/tests/unittest/llmapi/apps/_test_openai_chat_harmony.py b/tests/unittest/llmapi/apps/_test_openai_chat_harmony.py index ba6c7d5337..575cd2f0f1 100644 --- a/tests/unittest/llmapi/apps/_test_openai_chat_harmony.py +++ b/tests/unittest/llmapi/apps/_test_openai_chat_harmony.py @@ -14,10 +14,18 @@ def model(): return "gpt_oss/gpt-oss-20b/" +@pytest.fixture(scope="module", + params=[0, 2], + ids=["disable_processpool", "enable_processpool"]) +def num_postprocess_workers(request): + return request.param + + @pytest.fixture(scope="module") -def server(model: str): +def server(model: str, num_postprocess_workers: int): model_path = get_model_path(model) - with RemoteOpenAIServer(model_path) as remote_server: + args = ["--num_postprocess_workers", f"{num_postprocess_workers}"] + with RemoteOpenAIServer(model_path, args) as remote_server: yield remote_server