diff --git a/tests/unit/providers/inference/test_remote_vllm.py b/tests/unit/providers/inference/test_remote_vllm.py index 88399198d..491b2ba2e 100644 --- a/tests/unit/providers/inference/test_remote_vllm.py +++ b/tests/unit/providers/inference/test_remote_vllm.py @@ -107,8 +107,8 @@ async def test_register_model_checks_vllm(mock_openai_models_list, vllm_inferenc foo_model = Model(identifier="foo", provider_resource_id="foo", provider_id="vllm-inference") - await vllm_inference_adapter.register_model(foo_model) - mock_openai_models_list.assert_called() + with pytest.raises(ValueError): + await vllm_inference_adapter.register_model(foo_model) @pytest.mark.asyncio