From d814184bc36f48fa6ab1027bfeece225cc6365d0 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Wed, 6 Dec 2023 17:21:47 -0800 Subject: [PATCH] test: fix test imports --- litellm/proxy/custom_auth.py | 2 +- litellm/tests/test_configs/custom_auth.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/litellm/proxy/custom_auth.py b/litellm/proxy/custom_auth.py index 0cce561ca8..9334797082 100644 --- a/litellm/proxy/custom_auth.py +++ b/litellm/proxy/custom_auth.py @@ -1,4 +1,4 @@ -from litellm.proxy.types import UserAPIKeyAuth +from litellm.proxy._types import UserAPIKeyAuth from fastapi import Request from dotenv import load_dotenv import os diff --git a/litellm/tests/test_configs/custom_auth.py b/litellm/tests/test_configs/custom_auth.py index f9de3a97ab..b37ff8370a 100644 --- a/litellm/tests/test_configs/custom_auth.py +++ b/litellm/tests/test_configs/custom_auth.py @@ -1,4 +1,4 @@ -from litellm.proxy.types import UserAPIKeyAuth +from litellm.proxy._types import UserAPIKeyAuth from fastapi import Request from dotenv import load_dotenv import os