diff --git a/tests/router_unit_tests/test_router_cooldown_utils.py b/tests/router_unit_tests/test_router_cooldown_utils.py index 2a82b0d0915..53edb19de43 100644 --- a/tests/router_unit_tests/test_router_cooldown_utils.py +++ b/tests/router_unit_tests/test_router_cooldown_utils.py @@ -32,6 +32,7 @@ from litellm.router_utils.cooldown_handlers import _should_cooldown_deployment load_dotenv() + @pytest.mark.asyncio async def test_router_cooldown_event_callback_no_deployment(): """ @@ -410,4 +411,6 @@ def test_is_cooldown_required_empty_string_exception_status(testing_litellm_rout exception_status="", ) - assert result is False, "Should not require cooldown when exception_status is empty string" + assert ( + result is False + ), "Should not require cooldown when exception_status is empty string"