diff --git a/tests/test_litellm/llms/vertex_ai/test_vertex_passthrough_logging_handler.py b/tests/test_litellm/llms/vertex_ai/test_vertex_passthrough_logging_handler.py index 179301a9c81..2cf6689261e 100644 --- a/tests/test_litellm/llms/vertex_ai/test_vertex_passthrough_logging_handler.py +++ b/tests/test_litellm/llms/vertex_ai/test_vertex_passthrough_logging_handler.py @@ -158,6 +158,7 @@ def test_audio_predict_response_supports_bytes_base64_encoded( def test_lyria_predict_cost_falls_back_to_bundled_map_when_runtime_metadata_is_incomplete( monkeypatch: pytest.MonkeyPatch, missing_fields: tuple[str, ...] | None, + local_model_cost_map: None, ) -> None: if missing_fields is None: monkeypatch.delitem(litellm.model_cost, "vertex_ai/lyria-002")