[CI Bugfix] Fix failing test_invalid_env (#25078)

Signed-off-by: mgoin <mgoin64@gmail.com>
2025-09-17 11:28:58 -04:00
parent 4a2d33e371
commit 087c6ffc92
1 changed files with 1 additions and 1 deletions
--- 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)