[CI Bugfix] Fix failing test_invalid_env (#25078)

Signed-off-by: mgoin <mgoin64@gmail.com>

[CI Bugfix] Fix failing test_invalid_env (#25078)
Signed-off-by: mgoin <mgoin64@gmail.com>
087c6ffc · Michael Goin · GitHub · 4a2d33e3 · 087c6ffc
Unverified Commit 087c6ffc authored Sep 17, 2025 by Michael Goin Committed by GitHub Sep 17, 2025
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

tests/kernels/attention/test_attention_selector.py tests/kernels/attention/test_attention_selector.py +1 -1

No files found.
--- a/tests/kernels/attention/test_attention_selector.py
+++ b/tests/kernels/attention/test_attention_selector.py
@@ -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)