From 13e79fc8111b9eb3a2a5a367ea08f5d7fbf57281 Mon Sep 17 00:00:00 2001
From: Angela Yi <yiangela7@gmail.com>
Date: Wed, 11 Mar 2026 03:08:16 -0700
Subject: [PATCH] [ci] Update rtol for test_classification (#36556)

Signed-off-by: angelayi <yiangela7@gmail.com>
Co-authored-by: Richard Zou <zou3519@users.noreply.github.com>
---
 tests/models/language/pooling/test_classification.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/tests/models/language/pooling/test_classification.py b/tests/models/language/pooling/test_classification.py
index 2723bb21de9..e7128197bfc 100644
--- 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,
         )