Skip to content

Commit 087c6ff

Browse files
authored
[CI Bugfix] Fix failing test_invalid_env (#25078)
Signed-off-by: mgoin <mgoin64@gmail.com>
1 parent 4a2d33e commit 087c6ff

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/kernels/attention/test_attention_selector.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -368,4 +368,4 @@ def test_invalid_env(use_v1: bool, monkeypatch: pytest.MonkeyPatch):
368368
# Should raise ValueError for invalid backend
369369
with pytest.raises(ValueError) as exc_info:
370370
get_attn_backend(32, torch.float16, None, 16, False)
371-
assert "Invalid attention backend: 'INVALID'" in str(exc_info.value)
371+
assert "Invalid value 'INVALID'" in str(exc_info.value)

0 commit comments

Comments
 (0)