phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-27 03:34:10 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix: fix learnlm test

Browse source
This commit is contained in:
Krrish Dholakia 2025-03-13 10:54:09 -07:00
parent 88dbfc489d
commit a2f8e1a2a3
3 changed files with 7 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
tests/local_testing/test_get_model_info.py
View file

@ -121,7 +121,11 @@ def test_get_model_info_gemini():
model_map = litellm.model_cost
for model, info in model_map.items():
if model.startswith("gemini/") and not "gemma" in model:
if (
model.startswith("gemini/")
and not "gemma" in model
and not "learnlm" in model
):
assert info.get("tpm") is not None, f"{model} does not have tpm"
assert info.get("rpm") is not None, f"{model} does not have rpm"