From 4828fb1544bd555b5fdb7dce88ebe12b8dfa29f3 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Tue, 18 Feb 2025 21:29:23 -0800 Subject: [PATCH] fix code quality --- litellm/router_strategy/budget_limiter.py | 7 +------ 1 file changed, 1 insertion(+), 6 deletions(-) diff --git a/litellm/router_strategy/budget_limiter.py b/litellm/router_strategy/budget_limiter.py index 8061aab5376..4f123df2825 100644 --- a/litellm/router_strategy/budget_limiter.py +++ b/litellm/router_strategy/budget_limiter.py @@ -33,12 +33,7 @@ from litellm.router_utils.cooldown_callbacks import ( _get_prometheus_logger_from_callbacks, ) from litellm.types.llms.openai import AllMessageValues -from litellm.types.router import ( - DeploymentTypedDict, - LiteLLM_Params, - ModelInfo, - RouterErrors, -) +from litellm.types.router import DeploymentTypedDict, LiteLLM_Params, RouterErrors from litellm.types.utils import BudgetConfig from litellm.types.utils import BudgetConfig as GenericBudgetInfo from litellm.types.utils import GenericBudgetConfigType, StandardLoggingPayload