From 8bf5601ceb6f5f96869e0541e6d2e9efa5bafe9a Mon Sep 17 00:00:00 2001 From: Ishaan Jaffer Date: Sat, 21 Feb 2026 11:27:19 -0800 Subject: [PATCH] fix(tests): restore litellm.callbacks in TestSpendLogsPayload setup/teardown --- .../proxy/spend_tracking/test_spend_management_endpoints.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/tests/test_litellm/proxy/spend_tracking/test_spend_management_endpoints.py b/tests/test_litellm/proxy/spend_tracking/test_spend_management_endpoints.py index d05645b6e9d..e5d805f04c7 100644 --- a/tests/test_litellm/proxy/spend_tracking/test_spend_management_endpoints.py +++ b/tests/test_litellm/proxy/spend_tracking/test_spend_management_endpoints.py @@ -1211,6 +1211,12 @@ async def _wait_for_mock_call(mock, timeout=10, interval=0.1): class TestSpendLogsPayload: + def setup_method(self): + self._original_callbacks = litellm.callbacks[:] + + def teardown_method(self): + litellm.callbacks = self._original_callbacks + @pytest.mark.asyncio async def test_spend_logs_payload_e2e(self): litellm.callbacks = [_ProxyDBLogger(message_logging=False)]