llama_toolchain/distribution -> llama_toolchain/core

This commit is contained in:
Ashwin Bharambe 2024-08-28 17:39:41 -07:00
parent 81540e6ce8
commit 3cb67f1f58
31 changed files with 49 additions and 45 deletions

View file

@ -8,13 +8,13 @@ import argparse
from typing import Dict
from llama_toolchain.cli.subcommand import Subcommand
from llama_toolchain.distribution.datatypes import * # noqa: F403
from llama_toolchain.core.datatypes import * # noqa: F403
def parse_dependencies(
dependencies: str, parser: argparse.ArgumentParser
) -> Dict[str, ProviderSpec]:
from llama_toolchain.distribution.distribution import api_providers
from llama_toolchain.core.distribution import api_providers
all_providers = api_providers()
@ -48,8 +48,8 @@ class ApiBuild(Subcommand):
self.parser.set_defaults(func=self._run_api_build_command)
def _add_arguments(self):
from llama_toolchain.distribution.distribution import stack_apis
from llama_toolchain.distribution.package import (
from llama_toolchain.core.distribution import stack_apis
from llama_toolchain.core.package import (
BuildType,
)
@ -86,7 +86,7 @@ class ApiBuild(Subcommand):
)
def _run_api_build_command(self, args: argparse.Namespace) -> None:
from llama_toolchain.distribution.package import (
from llama_toolchain.core.package import (
ApiInput,
BuildType,
build_package,