From a43c792cfd39c74896cba89a1dc899dbde171f9b Mon Sep 17 00:00:00 2001 From: slalom Date: Fri, 29 May 2026 14:52:34 -0400 Subject: [PATCH] reorg --- src/commands/infra.py | 13 ++++--------- 1 file changed, 4 insertions(+), 9 deletions(-) diff --git a/src/commands/infra.py b/src/commands/infra.py index 9ca84ff..a27ef9b 100644 --- a/src/commands/infra.py +++ b/src/commands/infra.py @@ -154,7 +154,10 @@ def status(config: str = CONFIG_OPT) -> None: def mlflow_url(config: str = CONFIG_OPT) -> None: """Print a presigned URL for the configured MLflow tracking server.""" cfg = load_cfg(config) - tracking_server_name = _mlflow_tracking_server_name(cfg) + tracking_server_name = cfg.effective_mlflow_tracking_server_name + if not tracking_server_name: + CONSOLE.print("[red]MLflow is disabled in config.yaml.[/red]") + raise typer.Exit(1) try: url = mlflow.create_presigned_tracking_server_url( @@ -237,14 +240,6 @@ def _role_name(configured_name: str, role_arn: str) -> str: return "-" -def _mlflow_tracking_server_name(cfg: Config) -> str: - name = cfg.effective_mlflow_tracking_server_name - if not name: - CONSOLE.print("[red]MLflow is disabled in config.yaml.[/red]") - raise typer.Exit(1) - return name - - def _destroy_account_id(config_path: str, cfg: Config) -> str: config_dir = str(Path(config_path).parent) state = read_infra_state(config_dir)