Skip to content

Commit ec54d73

Browse files
authored
[CI] Fix test_collective_rpc (#17858)
Signed-off-by: Russell Bryant <rbryant@redhat.com>
1 parent a944f8e commit ec54d73

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

tests/entrypoints/llm/test_collective_rpc.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@
1010
@pytest.mark.parametrize("tp_size", [1, 2])
1111
@pytest.mark.parametrize("backend", ["mp", "ray"])
1212
@create_new_process_for_each_test()
13-
def test_collective_rpc(tp_size, backend):
13+
def test_collective_rpc(tp_size, backend, monkeypatch):
1414
if tp_size == 1 and backend == "ray":
1515
pytest.skip("Skip duplicate test case")
1616
if tp_size == 1:
@@ -21,6 +21,7 @@ def test_collective_rpc(tp_size, backend):
2121
def echo_rank(self):
2222
return self.rank
2323

24+
monkeypatch.setenv("VLLM_ALLOW_INSECURE_SERIALIZATION", "1")
2425
llm = LLM(model="meta-llama/Llama-3.2-1B-Instruct",
2526
enforce_eager=True,
2627
load_format="dummy",

0 commit comments

Comments
 (0)