diff --git a/llama_toolchain/cli/stack/build.py b/llama_toolchain/cli/stack/build.py index fb849335f..54eb1044c 100644 --- a/llama_toolchain/cli/stack/build.py +++ b/llama_toolchain/cli/stack/build.py @@ -68,7 +68,6 @@ class StackBuild(Subcommand): from llama_toolchain.common.config_dirs import DISTRIBS_BASE_DIR from llama_toolchain.common.serialize import EnumEncoder - from llama_toolchain.core.distribution_registry import resolve_distribution_spec from llama_toolchain.core.package import ApiInput, build_package, ImageType from termcolor import cprint diff --git a/llama_toolchain/core/distribution_registry.py b/llama_toolchain/core/distribution_registry.py index abaca3ab8..5af62bfde 100644 --- a/llama_toolchain/core/distribution_registry.py +++ b/llama_toolchain/core/distribution_registry.py @@ -57,13 +57,3 @@ def available_distribution_specs() -> List[DistributionSpec]: }, ), ] - - -@lru_cache() -def resolve_distribution_spec( - distribution_type: str, -) -> Optional[DistributionSpec]: - for spec in available_distribution_specs(): - if spec.distribution_type == distribution_type: - return spec - return None