From b9bddac77691747b3130c2de341f05e42891900d Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Mon, 3 Mar 2025 13:33:39 -0800 Subject: [PATCH] test: fix test --- tests/local_testing/test_get_model_info.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tests/local_testing/test_get_model_info.py b/tests/local_testing/test_get_model_info.py index c879332c7b5..89ebcfaff01 100644 --- a/tests/local_testing/test_get_model_info.py +++ b/tests/local_testing/test_get_model_info.py @@ -340,6 +340,8 @@ def test_get_model_info_bedrock_models(): base_model = BedrockModelInfo.get_base_model(k) base_model_info = litellm.model_cost[base_model] for base_model_key, base_model_value in base_model_info.items(): + if "invoke/" in k: + continue if base_model_key.startswith("supports_"): assert ( base_model_key in v