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)