Small refactor for run_with_pty

This commit is contained in:
Ashwin Bharambe 2025-01-28 09:32:33 -08:00
parent 8332ea23ad
commit aee6237685
3 changed files with 13 additions and 10 deletions

View file

@ -6,7 +6,6 @@
import argparse
import os
import sys
from pathlib import Path
from llama_stack.cli.subcommand import Subcommand
@ -71,7 +70,7 @@ class StackRun(Subcommand):
BUILDS_BASE_DIR,
DISTRIBS_BASE_DIR,
)
from llama_stack.distribution.utils.exec import run_with_pty, run_with_win
from llama_stack.distribution.utils.exec import run_with_pty
if not args.config:
self.parser.error("Must specify a config file to run")
@ -199,4 +198,4 @@ class StackRun(Subcommand):
return
run_args.extend(["--env", f"{key}={value}"])
run_with_win(args) if sys.platform.startswith("win") else run_with_pty(args)
run_with_pty(run_args)