[WIP][CI][Bugfix] Fix test_run_eagle_dp (#38584)

Signed-off-by: Matthew Bonanni <mbonanni@redhat.com>
2026-03-31 06:30:25 -04:00
parent 8278825b57
commit 7d65463528
2 changed files with 6 additions and 5 deletions
--- a/tests/v1/distributed/test_eagle_dp.py
+++ b/tests/v1/distributed/test_eagle_dp.py
@@ -69,9 +69,7 @@ async def test_run_eagle_dp(monkeypatch: pytest.MonkeyPatch, attn_backend: str):
    )

    prompt = "This is a test of data parallel with eagle"
-    # This test might be flaky, see
-    # https://github.com/vllm-project/vllm/issues/31913
-    num_expected_tokens = 20
+    num_expected_tokens = 100
    sampling_params = SamplingParams(
        max_tokens=num_expected_tokens,
        ignore_eos=True,