diff --git a/tests/unit/server/test_resolver.py b/tests/unit/server/test_resolver.py index 3af9535a0..821ddf1fe 100644 --- a/tests/unit/server/test_resolver.py +++ b/tests/unit/server/test_resolver.py @@ -19,7 +19,7 @@ from llama_stack.distribution.datatypes import ( StackRunConfig, ) from llama_stack.distribution.resolver import resolve_impls -from llama_stack.distribution.routers.routers import InferenceRouter +from llama_stack.distribution.routers.inference import InferenceRouter from llama_stack.distribution.routers.routing_tables import ModelsRoutingTable from llama_stack.providers.datatypes import InlineProviderSpec, ProviderSpec