mirror of
https://github.com/BerriAI/litellm.git
synced 2026-10-03 02:22:24 +00:00
fix(tests): guard conftest backfill against empty local cost map
This commit is contained in:
parent
99fda58f57
commit
3c4e1c21ae
1 changed files with 3 additions and 2 deletions
|
|
@ -31,9 +31,10 @@ import litellm
|
|||
# the cassette state the branch is being tested with.
|
||||
from litellm.litellm_core_utils.get_model_cost_map import GetModelCostMap
|
||||
|
||||
for _k, _v in GetModelCostMap.load_local_model_cost_map().items():
|
||||
_local_cost_map = GetModelCostMap.load_local_model_cost_map()
|
||||
for _k, _v in _local_cost_map.items():
|
||||
litellm.model_cost.setdefault(_k, _v)
|
||||
del _k, _v
|
||||
del _local_cost_map
|
||||
|
||||
from tests._vcr_conftest_common import ( # noqa: E402,F401
|
||||
VerboseReporterState,
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue