From 318142a17dbd2a04325c3c7a215e3033291889c8 Mon Sep 17 00:00:00 2001 From: Xi Yan Date: Sat, 14 Sep 2024 14:24:29 -0700 Subject: [PATCH] remove available_distribution_specs --- llama_toolchain/cli/stack/build.py | 6 ------ llama_toolchain/cli/stack/configure.py | 6 ------ 2 files changed, 12 deletions(-) diff --git a/llama_toolchain/cli/stack/build.py b/llama_toolchain/cli/stack/build.py index 54eb1044c..8448a45ef 100644 --- a/llama_toolchain/cli/stack/build.py +++ b/llama_toolchain/cli/stack/build.py @@ -48,12 +48,6 @@ class StackBuild(Subcommand): self.parser.set_defaults(func=self._run_stack_build_command) def _add_arguments(self): - from llama_toolchain.core.distribution_registry import ( - available_distribution_specs, - ) - from llama_toolchain.core.package import ImageType - - allowed_ids = [d.distribution_type for d in available_distribution_specs()] self.parser.add_argument( "--config", type=str, diff --git a/llama_toolchain/cli/stack/configure.py b/llama_toolchain/cli/stack/configure.py index bd1466874..d48ba47a4 100644 --- a/llama_toolchain/cli/stack/configure.py +++ b/llama_toolchain/cli/stack/configure.py @@ -32,12 +32,6 @@ class StackConfigure(Subcommand): self.parser.set_defaults(func=self._run_stack_configure_cmd) def _add_arguments(self): - from llama_toolchain.core.distribution_registry import ( - available_distribution_specs, - ) - from llama_toolchain.core.package import ImageType - - allowed_ids = [d.distribution_type for d in available_distribution_specs()] self.parser.add_argument( "config", type=str,