phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

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 84285b9dbb
commit f1b0cc756b
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()