diff --git a/tests/router_unit_tests/test_router_helper_utils.py b/tests/router_unit_tests/test_router_helper_utils.py index 45aae3b9aee..073433cb9e5 100644 --- a/tests/router_unit_tests/test_router_helper_utils.py +++ b/tests/router_unit_tests/test_router_helper_utils.py @@ -73,7 +73,7 @@ def test_routing_strategy_init(model_list): from litellm.types.router import RoutingStrategy router = Router(model_list=model_list) - for strategy in RoutingStrategy._member_names_: + for strategy in RoutingStrategy: router.routing_strategy_init( routing_strategy=strategy, routing_strategy_args={} )