diff --git a/tests/integration/providers/test_image_gen_drop_params_wire.py b/tests/integration/providers/test_image_gen_drop_params_wire.py index c9cfc30e24d..f61d7d6dc17 100644 --- a/tests/integration/providers/test_image_gen_drop_params_wire.py +++ b/tests/integration/providers/test_image_gen_drop_params_wire.py @@ -82,6 +82,7 @@ async def test_proxy_image_retry_attempts_are_owned_only_by_router( ) finally: release.set() + router.discard() litellm.in_memory_llm_clients_cache.flush_cache() requests: Final = wire.drain() status: Final = {"503": 503, "timeout": 408, "success": 200}[outcome]