From cf897d188be2968b06948b1d35580fd5801a2e89 Mon Sep 17 00:00:00 2001 From: Xi Yan Date: Sat, 14 Sep 2024 15:03:15 -0700 Subject: [PATCH] generate distribution specs --- .../distributions/{run.py => generate_distribution_specs.py} | 0 llama_toolchain/core/distribution_registry.py | 2 +- 2 files changed, 1 insertion(+), 1 deletion(-) rename llama_toolchain/configs/distributions/{run.py => generate_distribution_specs.py} (100%) diff --git a/llama_toolchain/configs/distributions/run.py b/llama_toolchain/configs/distributions/generate_distribution_specs.py similarity index 100% rename from llama_toolchain/configs/distributions/run.py rename to llama_toolchain/configs/distributions/generate_distribution_specs.py diff --git a/llama_toolchain/core/distribution_registry.py b/llama_toolchain/core/distribution_registry.py index 0c3e5a90c..87ac98d7c 100644 --- a/llama_toolchain/core/distribution_registry.py +++ b/llama_toolchain/core/distribution_registry.py @@ -11,7 +11,7 @@ from .datatypes import * # noqa: F403 import yaml -# @lru_cache() +@lru_cache() def available_distribution_specs() -> List[DistributionSpec]: distribution_specs = [] for p in Path("llama_toolchain/configs/distributions/distribution_registry").rglob(