From a79079feef63d7a8124782cf2eb1cd68cbde99ee Mon Sep 17 00:00:00 2001 From: Richard Zou Date: Wed, 7 Jan 2026 23:04:58 -0500 Subject: [PATCH] [BugFix] Fix flakiness in test_eagle_dp for PyTorch 2.10 (#31915) Signed-off-by: Richard Zou --- tests/v1/distributed/test_eagle_dp.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/tests/v1/distributed/test_eagle_dp.py b/tests/v1/distributed/test_eagle_dp.py index f529fce0ab0..f62e85dcdaf 100644 --- 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,