We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 88e129b commit 6bfa0e3Copy full SHA for 6bfa0e3
tests/kernels/attention/test_attention_selector.py
@@ -368,4 +368,4 @@ def test_invalid_env(use_v1: bool, monkeypatch: pytest.MonkeyPatch):
368
# Should raise ValueError for invalid backend
369
with pytest.raises(ValueError) as exc_info:
370
get_attn_backend(32, torch.float16, None, 16, False)
371
- assert "Invalid attention backend: 'INVALID'" in str(exc_info.value)
+ assert "Invalid value 'INVALID'" in str(exc_info.value)
0 commit comments