From 074c8efdc1805fa5a5cc449bc3dd76be9dbbfb96 Mon Sep 17 00:00:00 2001 From: Utkash Dubey Date: Mon, 10 Mar 2025 20:13:03 -0700 Subject: [PATCH] cleanup --- tests/litellm/test_model_prices_and_context_window_schema.py | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/tests/litellm/test_model_prices_and_context_window_schema.py b/tests/litellm/test_model_prices_and_context_window_schema.py index 3882289d9b..085512da88 100644 --- a/tests/litellm/test_model_prices_and_context_window_schema.py +++ b/tests/litellm/test_model_prices_and_context_window_schema.py @@ -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: actual_json = json.load(model_prices_file) 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) - if temporarily_removed is not None: - # put back the sample spec that we removed - actual_json.update({'sample_spec': temporarily_removed})