[CI] Fix test_async_scheduling.py flakiness (#42455)

Signed-off-by: Nick Hill <nickhill123@gmail.com>
2026-05-12 21:38:32 +00:00
parent fe5b4e0fe7
commit fe8b42e80c
1 changed files with 1 additions and 1 deletions
@@ -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)
        )