From fe8b42e80c482ea6d93bc5f800109528f2f11e31 Mon Sep 17 00:00:00 2001 From: Nick Hill Date: Tue, 12 May 2026 14:38:32 -0700 Subject: [PATCH] [CI] Fix `test_async_scheduling.py` flakiness (#42455) Signed-off-by: Nick Hill --- tests/v1/e2e/general/test_async_scheduling.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/v1/e2e/general/test_async_scheduling.py b/tests/v1/e2e/general/test_async_scheduling.py index c3c4970de38..22a6c799c79 100644 --- a/tests/v1/e2e/general/test_async_scheduling.py +++ b/tests/v1/e2e/general/test_async_scheduling.py @@ -429,7 +429,7 @@ def _logprobs_match( and lps_a.keys() == lps_b.keys() and all( a.decoded_token == b.decoded_token - and a.rank == b.rank + and a.rank == pytest.approx(b.rank, rel=0.005) and a.logprob == pytest.approx(b.logprob, rel=rel_tol, abs=abs_tol) for a, b in ((lps_a[x], lps_b[x]) for x in lps_a) )