From ec54d73c3105ccfdd9b036d1ca0feb62078e20f7 Mon Sep 17 00:00:00 2001 From: Russell Bryant Date: Thu, 8 May 2025 12:47:12 -0400 Subject: [PATCH] [CI] Fix test_collective_rpc (#17858) Signed-off-by: Russell Bryant --- tests/entrypoints/llm/test_collective_rpc.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tests/entrypoints/llm/test_collective_rpc.py b/tests/entrypoints/llm/test_collective_rpc.py index d51b7c26344f..6470249dddbc 100644 --- a/tests/entrypoints/llm/test_collective_rpc.py +++ b/tests/entrypoints/llm/test_collective_rpc.py @@ -10,7 +10,7 @@ from ...utils import create_new_process_for_each_test @pytest.mark.parametrize("tp_size", [1, 2]) @pytest.mark.parametrize("backend", ["mp", "ray"]) @create_new_process_for_each_test() -def test_collective_rpc(tp_size, backend): +def test_collective_rpc(tp_size, backend, monkeypatch): if tp_size == 1 and backend == "ray": pytest.skip("Skip duplicate test case") if tp_size == 1: @@ -21,6 +21,7 @@ def test_collective_rpc(tp_size, backend): def echo_rank(self): return self.rank + monkeypatch.setenv("VLLM_ALLOW_INSECURE_SERIALIZATION", "1") llm = LLM(model="meta-llama/Llama-3.2-1B-Instruct", enforce_eager=True, load_format="dummy",