From f05347be8f7a7f9e6432e0489de09d602e09f1ff Mon Sep 17 00:00:00 2001 From: Xi Yan Date: Tue, 10 Sep 2024 11:15:38 -0700 Subject: [PATCH] fix configure script to work with config file --- llama_toolchain/cli/stack/configure.py | 13 +++++++------ 1 file changed, 7 insertions(+), 6 deletions(-) diff --git a/llama_toolchain/cli/stack/configure.py b/llama_toolchain/cli/stack/configure.py index 716169177..f90f3ba0f 100644 --- a/llama_toolchain/cli/stack/configure.py +++ b/llama_toolchain/cli/stack/configure.py @@ -63,17 +63,18 @@ class StackConfigure(Subcommand): from llama_toolchain.core.package import BuildType if args.config_file: - build_config = BuildConfig(**yaml.safe_load(f)) - build_type = BuildType(build_config.package_type) + with open(args.config_file, "r") as f: + build_config = BuildConfig(**yaml.safe_load(f)) + build_type = BuildType(build_config.package_type) + distribution = build_config.distribution + name = build_config.name else: build_type = BuildType(args.package_type) name = args.name + distribution = args.distribution config_file = ( - BUILDS_BASE_DIR - / args.distribution - / build_type.descriptor() - / f"{name}.yaml" + BUILDS_BASE_DIR / distribution / build_type.descriptor() / f"{name}.yaml" ) if not config_file.exists(): self.parser.error(