Unverified Commit 8b9e8b74 authored by Micah Williamson's avatar Micah Williamson Committed by GitHub
Browse files

[ROCm][CI] Fix Assertion Logic For `test_gpt_oss` (#35806)


Signed-off-by: default avatarMicah Williamson <micah.williamson@amd.com>
parent c21d0039
......@@ -12,8 +12,8 @@ Config:
Run: pytest tests/models/quantization/test_gpt_oss.py
"""
import importlib
import importlib.metadata
import importlib.util
from dataclasses import dataclass
import huggingface_hub
......@@ -104,7 +104,7 @@ def test_gpt_oss_attention_quantization(
)
rtol = 0.02
assert (
measured_accuracy - rtol < expected_accuracy
and measured_accuracy + rtol > expected_accuracy
), f"Expected: {expected_accuracy} | Measured: {measured_accuracy}"
assert measured_accuracy >= expected_accuracy - rtol, (
f"Accuracy {measured_accuracy:.4f} is below threshold "
f"{expected_accuracy - rtol:.4f} (expected >= {expected_accuracy} - {rtol})"
)
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment