[CI Bugfix] Fix failing test_invalid_env (#25078)
Signed-off-by: mgoin <mgoin64@gmail.com>
This commit is contained in:
@@ -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)
|
||||||
|
|||||||
Reference in New Issue
Block a user