From 18c5265f8f928b354e46e09fb705ff42944be4e3 Mon Sep 17 00:00:00 2001 From: einsmein Date: Mon, 6 Dec 2021 17:31:03 +0100 Subject: [PATCH] Assign run id when saving a model (#4256) Signed-off-by: Mind Co-authored-by: Mind --- mlflow/R/mlflow/R/model.R | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mlflow/R/mlflow/R/model.R b/mlflow/R/mlflow/R/model.R index 0204d20d83c03..b7125b7aaada7 100644 --- a/mlflow/R/mlflow/R/model.R +++ b/mlflow/R/mlflow/R/model.R @@ -31,7 +31,7 @@ mlflow_log_model <- function(model, artifact_path, ...) { temp_path <- fs::path_temp(artifact_path) model_spec <- mlflow_save_model(model, path = temp_path, model_spec = list( utc_time_created = mlflow_timestamp(), - run_id = mlflow_get_active_run_id(), + run_id = mlflow_get_active_run_id_or_start_run(), artifact_path = artifact_path, flavors = list() ), ...)