[CI Bugfix] Fix failing test_invalid_env (#25078)

Signed-off-by: mgoin <mgoin64@gmail.com>
This commit is contained in:
Michael Goin 2025-09-17 11:28:58 -04:00 committed by GitHub
parent 4a2d33e371
commit 087c6ffc92
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -368,4 +368,4 @@ def test_invalid_env(use_v1: bool, monkeypatch: pytest.MonkeyPatch):
# Should raise ValueError for invalid backend
with pytest.raises(ValueError) as exc_info:
get_attn_backend(32, torch.float16, None, 16, False)
assert "Invalid attention backend: 'INVALID'" in str(exc_info.value)
assert "Invalid value 'INVALID'" in str(exc_info.value)