diff --git a/litellm/router.py b/litellm/router.py index 1a79bedaf..9c56445d3 100644 --- a/litellm/router.py +++ b/litellm/router.py @@ -311,7 +311,7 @@ class Router: ) # noqa self.routing_strategy_args = routing_strategy_args self.retry_policy: Optional[RetryPolicy] = retry_policy - self.model_group_retry_policy: Optional[dict[str, RetryPolicy]] = ( + self.model_group_retry_policy: Optional[Dict[str, RetryPolicy]] = ( model_group_retry_policy )