Skip to content

Commit fb6cfc7

Browse files
hmelloradobrzyn
authored andcommitted
Fix OOT registration test (vllm-project#17099)
Signed-off-by: Harry Mellor <19981378+hmellor@users.noreply.github.com> Signed-off-by: Agata Dobrzyniewicz <adobrzyniewicz@habana.ai>
1 parent 76d36be commit fb6cfc7

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

tests/models/test_oot_registration.py

+2-3
Original file line numberDiff line numberDiff line change
@@ -18,10 +18,9 @@ def test_plugin(
1818
m.setenv("VLLM_USE_V1", "0")
1919
m.setenv("VLLM_PLUGINS", "")
2020

21-
with pytest.raises(Exception) as excinfo:
21+
match = "Cannot find model module"
22+
with pytest.raises(ValueError, match=match):
2223
LLM(model=dummy_opt_path, load_format="dummy")
23-
error_msg = "has no vLLM implementation and the Transformers implementation is not compatible with vLLM" # noqa: E501
24-
assert (error_msg in str(excinfo.value))
2524

2625

2726
@create_new_process_for_each_test()

0 commit comments

Comments
 (0)