From d94065ca43d90aea7055f824032386e8059bd34d Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Mon, 6 May 2024 18:20:27 -0700 Subject: [PATCH] fix - num active callbacks --- tests/test_callbacks_on_proxy.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/test_callbacks_on_proxy.py b/tests/test_callbacks_on_proxy.py index a4e4fe7b8..ea15af6b0 100644 --- a/tests/test_callbacks_on_proxy.py +++ b/tests/test_callbacks_on_proxy.py @@ -111,12 +111,12 @@ async def test_check_num_callbacks(): import uuid async with aiohttp.ClientSession() as session: - # await asyncio.sleep(30) + await asyncio.sleep(30) num_callbacks_1, _, all_litellm_callbacks_1 = await get_active_callbacks( session=session ) assert num_callbacks_1 > 0 - # await asyncio.sleep(30) + await asyncio.sleep(30) num_callbacks_2, _, all_litellm_callbacks_2 = await get_active_callbacks( session=session @@ -166,7 +166,7 @@ async def test_check_num_callbacks_on_lowest_latency(): original_routing_strategy = await get_current_routing_strategy(session=session) await config_update(session=session, routing_strategy="latency-based-routing") - await asyncio.sleep(5) + await asyncio.sleep(30) num_callbacks_1, num_alerts_1, all_litellm_callbacks_1 = ( await get_active_callbacks(session=session)