refactor: fix imports

This commit is contained in:
Krrish Dholakia 2024-09-28 13:28:01 -07:00
parent 3560f0ef2c
commit 5ad01e59f6
8 changed files with 19 additions and 15 deletions

View file

@ -235,7 +235,7 @@ async def test_team_disable_guardrails(mock_acompletion, client_no_auth):
assert e.code == str(403)
from litellm.tests.test_custom_callback_input import CompletionCustomHandler
from tests.local_testing.test_custom_callback_input import CompletionCustomHandler
@mock_patch_acompletion()
@ -815,7 +815,7 @@ from litellm.proxy._types import (
)
from litellm.proxy.management_endpoints.internal_user_endpoints import new_user
from litellm.proxy.management_endpoints.team_endpoints import team_member_add
from litellm.tests.test_key_generate_prisma import prisma_client
from tests.local_testing.test_key_generate_prisma import prisma_client
@pytest.mark.parametrize(