Skip to content

GitLab

Unverified Commit 087c6ffc authored by Michael Goin's avatar Michael Goin Committed by GitHub
Browse files

[CI Bugfix] Fix failing test_invalid_env (#25078) 


Signed-off-by: default avatarmgoin <mgoin64@gmail.com>
parent 4a2d33e3
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tests/kernels/attention/test_attention_selector.py
View file @ 087c6ffc
...@@ -368,4 +368,4 @@ def test_invalid_env(use_v1: bool, monkeypatch: pytest.MonkeyPatch): ...@@ -368,4 +368,4 @@ def test_invalid_env(use_v1: bool, monkeypatch: pytest.MonkeyPatch):
# Should raise ValueError for invalid backend # Should raise ValueError for invalid backend
with pytest.raises(ValueError) as exc_info: with pytest.raises(ValueError) as exc_info:
get_attn_backend(32, torch.float16, None, 16, False) 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)
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