From e477b7e3daf8a24a7dee4e8af027d8be2e654076 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Sat, 6 Jul 2024 11:27:58 -0700 Subject: [PATCH] fix linting error --- litellm/tests/test_config.py | 22 +++++++++++++++------- 1 file changed, 15 insertions(+), 7 deletions(-) diff --git a/litellm/tests/test_config.py b/litellm/tests/test_config.py index cd61101a3c..28d144e4dc 100644 --- a/litellm/tests/test_config.py +++ b/litellm/tests/test_config.py @@ -2,23 +2,30 @@ ## Unit tests for ProxyConfig class -import sys, os +import os +import sys import traceback + from dotenv import load_dotenv load_dotenv() -import os, io +import io +import os sys.path.insert( 0, os.path.abspath("../..") ) # Adds the parent directory to the system path -import pytest, litellm -from pydantic import BaseModel, ConfigDict -from litellm.proxy.proxy_server import ProxyConfig -from litellm.proxy.utils import encrypt_value, ProxyLogging, DualCache -from litellm.types.router import Deployment, LiteLLM_Params, ModelInfo from typing import Literal +import pytest +from pydantic import BaseModel, ConfigDict + +import litellm +from litellm.proxy.common_utils.encrypt_decrypt_utils import encrypt_value +from litellm.proxy.proxy_server import ProxyConfig +from litellm.proxy.utils import DualCache, ProxyLogging +from litellm.types.router import Deployment, LiteLLM_Params, ModelInfo + class DBModel(BaseModel): model_id: str @@ -28,6 +35,7 @@ class DBModel(BaseModel): model_config = ConfigDict(protected_namespaces=()) + @pytest.mark.asyncio async def test_delete_deployment(): """