diff --git a/tests/unit/router_utils/test_fallback_event_handlers.py b/tests/unit/router_utils/test_fallback_event_handlers.py index 1cba3f8cab2..998f8308dd3 100644 --- a/tests/unit/router_utils/test_fallback_event_handlers.py +++ b/tests/unit/router_utils/test_fallback_event_handlers.py @@ -590,7 +590,7 @@ class TagRoutingFallbackRouter: def log_retry(self, kwargs, e): return kwargs - def _get_all_deployments(self, model_name: str, **kwargs): + def get_all_deployments(self, model_name: str, **kwargs): return list(self.deployments_by_group.get(model_name, [])) async def async_function_with_fallbacks(self, *args, **kwargs):