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

test(test_utils.py): reset env after test complete

Browse source
This commit is contained in:
Krrish Dholakia 2024-06-17 20:33:39 -07:00
parent a3e0267f1d
commit f47358ba1d
2 changed files with 5 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

5
litellm/tests/test_utils.py
View file

@ -200,7 +200,7 @@ def test_trimming_with_model_cost_max_input_tokens(model):
)
def test_get_valid_models():
def test_aget_valid_models():
old_environ = os.environ
os.environ = {"OPENAI_API_KEY": "temp"} # mock set only openai key in environ
@ -227,6 +227,9 @@ def test_get_valid_models():
print(valid_models)
assert valid_models == expected_models
# reset replicate env key
os.environ = old_environ
# test_get_valid_models()