Skip to content

GitLab

Unverified Commit ec54d73c authored by Russell Bryant's avatar Russell Bryant Committed by GitHub
Browse files

[CI] Fix test_collective_rpc (#17858) 


Signed-off-by: default avatarRussell Bryant <rbryant@redhat.com>
parent a944f8ed
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
tests/entrypoints/llm/test_collective_rpc.py
View file @ ec54d73c
...@@ -10,7 +10,7 @@ from ...utils import create_new_process_for_each_test ...@@ -10,7 +10,7 @@ from ...utils import create_new_process_for_each_test
@pytest.mark.parametrize("tp_size", [1, 2]) @pytest.mark.parametrize("tp_size", [1, 2])
@pytest.mark.parametrize("backend", ["mp", "ray"]) @pytest.mark.parametrize("backend", ["mp", "ray"])
@create_new_process_for_each_test() @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": if tp_size == 1 and backend == "ray":
pytest.skip("Skip duplicate test case") pytest.skip("Skip duplicate test case")
if tp_size == 1: if tp_size == 1:
...@@ -21,6 +21,7 @@ def test_collective_rpc(tp_size, backend): ...@@ -21,6 +21,7 @@ def test_collective_rpc(tp_size, backend):
def echo_rank(self): def echo_rank(self):
return self.rank return self.rank
monkeypatch.setenv("VLLM_ALLOW_INSECURE_SERIALIZATION", "1")
llm = LLM(model="meta-llama/Llama-3.2-1B-Instruct", llm = LLM(model="meta-llama/Llama-3.2-1B-Instruct",
enforce_eager=True, enforce_eager=True,
load_format="dummy", load_format="dummy",
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment