From b97b213c9901585cb71bd505779964007da4abe4 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Wed, 31 Jul 2024 11:48:36 -0700 Subject: [PATCH] fix test_team_disable_guardrails --- litellm/tests/test_proxy_server.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/litellm/tests/test_proxy_server.py b/litellm/tests/test_proxy_server.py index 0e5431c3fe..ecee1a40f5 100644 --- a/litellm/tests/test_proxy_server.py +++ b/litellm/tests/test_proxy_server.py @@ -227,7 +227,7 @@ async def test_team_disable_guardrails(mock_acompletion, client_no_auth): await user_api_key_auth(request=request, api_key="Bearer " + user_key) pytest.fail("Expected to raise 403 forbidden error.") except ProxyException as e: - assert e.code == 403 + assert e.code == "403" from litellm.tests.test_custom_callback_input import CompletionCustomHandler