diff --git a/tests/test_litellm/llms/bedrock_mantle/test_bedrock_mantle_responses_transformation.py b/tests/test_litellm/llms/bedrock_mantle/test_bedrock_mantle_responses_transformation.py index fd279a2bc1f..28c6060e5cc 100644 --- a/tests/test_litellm/llms/bedrock_mantle/test_bedrock_mantle_responses_transformation.py +++ b/tests/test_litellm/llms/bedrock_mantle/test_bedrock_mantle_responses_transformation.py @@ -1568,7 +1568,7 @@ class TestBedrockMantleResponsesPricing: assert "bedrock_mantle/openai.gpt-5.4" in litellm.bedrock_mantle_models -def _repo_cost_map(map_name: str) -> dict: +def _repo_cost_map(map_name: str) -> dict[str, dict[str, object]]: repo_root = Path(__file__).resolve().parents[4] paths = { "root": repo_root / "model_prices_and_context_window.json",