diff --git a/tests/test_litellm/test_utils.py b/tests/test_litellm/test_utils.py index 984c6b0117..35cb290fcc 100644 --- a/tests/test_litellm/test_utils.py +++ b/tests/test_litellm/test_utils.py @@ -808,8 +808,7 @@ def test_aaamodel_prices_and_context_window_json_is_valid(): }, } - prod_json = "litellm/model_prices_and_context_window.json" - # prod_json = "../../model_prices_and_context_window.json" + prod_json = os.path.join(os.path.dirname(__file__), "..", "..", "model_prices_and_context_window.json") with open(prod_json, "r") as model_prices_file: actual_json = json.load(model_prices_file) assert isinstance(actual_json, dict)