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)