phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-25 18:54:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

cleanup

Browse source
This commit is contained in:
Utkash Dubey 2025-03-10 20:13:03 -07:00
parent 14736283bc
commit 074c8efdc1
1 changed files with 1 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

5
tests/litellm/test_model_prices_and_context_window_schema.py
View file

@ -102,10 +102,7 @@ def test_model_prices_and_context_window_json_is_valid():
with open("./model_prices_and_context_window.json", "r") as model_prices_file: with open("./model_prices_and_context_window.json", "r") as model_prices_file:
actual_json = json.load(model_prices_file) actual_json = json.load(model_prices_file)
assert isinstance(actual_json, dict) assert isinstance(actual_json, dict)
temporarily_removed = actual_json.pop('sample_spec', None) # remove the sample, whose schema is inconsistent with the real data actual_json.pop('sample_spec', None) # remove the sample, whose schema is inconsistent with the real data
validate(actual_json, INTENDED_SCHEMA) validate(actual_json, INTENDED_SCHEMA)
if temporarily_removed is not None:
# put back the sample spec that we removed
actual_json.update({'sample_spec': temporarily_removed})