[BugFix] Fix flakiness in test_eagle_dp for PyTorch 2.10 (#31915)

Signed-off-by: Richard Zou <zou3519@gmail.com>
2026-01-07 23:04:58 -05:00
parent 9f6dcb71ae
commit a79079feef
1 changed files with 3 additions and 1 deletions
--- a/tests/v1/distributed/test_eagle_dp.py
+++ b/tests/v1/distributed/test_eagle_dp.py
@@ -47,7 +47,9 @@ async def test_run_eagle_dp(monkeypatch: pytest.MonkeyPatch):
    )

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