[ci] Update rtol for test_classification (#36556)

Signed-off-by: angelayi <yiangela7@gmail.com> Co-authored-by: Richard Zou <zou3519@users.noreply.github.com>
2026-03-11 03:08:16 -07:00
parent 9d07a3d6e4
commit 13e79fc811
1 changed files with 3 additions and 1 deletions
--- a/tests/models/language/pooling/test_classification.py
+++ b/tests/models/language/pooling/test_classification.py
@@ -45,5 +45,7 @@ def test_models(
        # half datatype tests in
        # tests/models/language/pooling/test_embedding.py
        assert torch.allclose(
-            hf_output, vllm_output, 1e-3 if dtype == "float" else 1e-2
+            hf_output,
+            vllm_output,
+            rtol=2e-3 if dtype == "float" else 1e-2,
        )