Skip to content

Commit b045658

Browse files
committed
refactor: remove container from list of run image types
1 parent 8e7ab14 commit b045658

File tree

2 files changed

+3
-4
lines changed

2 files changed

+3
-4
lines changed

llama_stack/cli/stack/run.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -57,7 +57,7 @@ def _add_arguments(self):
5757
"--image-type",
5858
type=str,
5959
help="Image Type used during the build. This can be either conda or container or venv.",
60-
choices=[e.value for e in ImageType],
60+
choices=[e.value for e in ImageType if e.value != "container"],
6161
)
6262

6363
# If neither image type nor image name is provided, but at the same time

llama_stack/distribution/utils/exec.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -22,9 +22,8 @@
2222

2323
def formulate_run_args(image_type, image_name, config, template_name) -> list:
2424
env_name = ""
25-
if image_type == LlamaStackImageType.CONTAINER.value or config.container_image:
26-
env_name = f"distribution-{template_name}" if template_name else config.container_image
27-
elif image_type == LlamaStackImageType.CONDA.value:
25+
26+
if image_type == LlamaStackImageType.CONDA.value:
2827
current_conda_env = os.environ.get("CONDA_DEFAULT_ENV")
2928
env_name = image_name or current_conda_env
3029
if not env_name:

0 commit comments

Comments
 (0)