omit server name when created with config

This commit is contained in:
2026-05-27 10:23:53 -04:00
parent 35d25d8967
commit e1c8d6574f
5 changed files with 30 additions and 14 deletions

View File

@@ -48,13 +48,14 @@ class MlflowTracker:
"Install with: qc-cli[mlflow]"
) from e
if not cfg.mlflow.tracking_server_name:
raise RuntimeError("mlflow.tracking_server_name is required when MLflow is enabled.")
tracking_server_name = cfg.effective_mlflow_tracking_server_name
if not tracking_server_name:
raise RuntimeError("MLflow tracking server name could not be resolved.")
tracking_uri = aws_mlflow.get_tracking_server_arn(
cfg.aws.region,
cfg.aws.profile,
cfg.mlflow.tracking_server_name,
tracking_server_name,
)
mlflow.set_tracking_uri(tracking_uri)
mlflow.set_experiment(cfg.mlflow.experiment_name)