mirror of
https://github.com/BerriAI/litellm.git
synced 2026-10-06 02:48:13 +00:00
fix(lint): remove unused imports in test_reset_budget_job.py
Clean up unused imports (time, verbose_proxy_logger, ProxyLogging, AsyncMock, LiteLLM_BudgetTableFull) to pass ruff F401 checks.
This commit is contained in:
parent
2ac8a2ee85
commit
bd4c57c0bb
1 changed files with 1 additions and 6 deletions
|
|
@ -1,7 +1,6 @@
|
|||
import asyncio
|
||||
import os
|
||||
import sys
|
||||
import time
|
||||
from datetime import datetime, timedelta, timezone
|
||||
from typing import Any, Dict, List
|
||||
|
||||
|
|
@ -11,9 +10,7 @@ sys.path.insert(
|
|||
0, os.path.abspath("../../..")
|
||||
) # Adds the parent directory to the system path
|
||||
|
||||
from litellm._logging import verbose_proxy_logger
|
||||
from litellm.proxy.common_utils.reset_budget_job import ResetBudgetJob
|
||||
from litellm.proxy.utils import ProxyLogging
|
||||
|
||||
|
||||
# Mock classes for testing
|
||||
|
|
@ -393,7 +390,7 @@ def test_enduser_budget_reset_reproduces_stale_cache_bug():
|
|||
This test simulates the full flow using real LiteLLM_EndUserTable and
|
||||
LiteLLM_BudgetTable Pydantic objects.
|
||||
"""
|
||||
from unittest.mock import AsyncMock, patch
|
||||
from unittest.mock import patch
|
||||
|
||||
from litellm.proxy._types import LiteLLM_BudgetTable, LiteLLM_EndUserTable
|
||||
|
||||
|
|
@ -596,8 +593,6 @@ def test_reset_budget_for_keys_linked_to_budgets(reset_budget_job, mock_prisma_c
|
|||
This covers the case where keys were created with budget_id but
|
||||
budget_duration was not inherited to the key (pre-fix keys).
|
||||
"""
|
||||
from litellm.proxy._types import LiteLLM_BudgetTableFull
|
||||
|
||||
now = datetime.now(timezone.utc)
|
||||
|
||||
# Create a budget tier that is due for reset
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue