diff --git a/tests/litellm/proxy/common_utils/test_integration_reset_enduser_spend.py b/tests/local_testing/test_enduser_spend_reset.py similarity index 99% rename from tests/litellm/proxy/common_utils/test_integration_reset_enduser_spend.py rename to tests/local_testing/test_enduser_spend_reset.py index 999584451b..832ba9b324 100644 --- a/tests/litellm/proxy/common_utils/test_integration_reset_enduser_spend.py +++ b/tests/local_testing/test_enduser_spend_reset.py @@ -10,16 +10,15 @@ import pytest_asyncio from dotenv import load_dotenv from litellm.caching.caching import DualCache - -load_dotenv() - -sys.path.insert( - 0, os.path.abspath("../../..") -) # Adds the parent directory to the system path - from litellm.proxy.common_utils.reset_budget_job import ResetBudgetJob from litellm.proxy.utils import PrismaClient, ProxyLogging +sys.path.insert( + 0, os.path.abspath("../..") +) # Adds the parent directory to the system path + +load_dotenv() + async def create_budget(session, data): url = "http://0.0.0.0:4000/budget/new"