Don't import pkg_resources until you need it

This commit is contained in:
Ashwin Bharambe 2024-09-17 20:01:22 -07:00
parent 25adc83de8
commit 3e27131a69
2 changed files with 4 additions and 3 deletions

View file

@ -8,8 +8,6 @@ import argparse
import json import json
from pathlib import Path from pathlib import Path
import pkg_resources
import yaml import yaml
from termcolor import cprint from termcolor import cprint
@ -49,6 +47,8 @@ class StackConfigure(Subcommand):
) )
def _run_stack_configure_cmd(self, args: argparse.Namespace) -> None: def _run_stack_configure_cmd(self, args: argparse.Namespace) -> None:
import pkg_resources
from llama_stack.distribution.build import ImageType from llama_stack.distribution.build import ImageType
docker_image = None docker_image = None

View file

@ -8,7 +8,6 @@ import argparse
from pathlib import Path from pathlib import Path
import pkg_resources
import yaml import yaml
from llama_stack.cli.subcommand import Subcommand from llama_stack.cli.subcommand import Subcommand
@ -47,6 +46,8 @@ class StackRun(Subcommand):
) )
def _run_stack_run_cmd(self, args: argparse.Namespace) -> None: def _run_stack_run_cmd(self, args: argparse.Namespace) -> None:
import pkg_resources
from llama_stack.distribution.utils.exec import run_with_pty from llama_stack.distribution.utils.exec import run_with_pty
if not args.config: if not args.config: