Assign run id when saving a model

mlflow · Apr 15, 2021 · f79a495 · f79a495
1 parent 6bbd93e
commit f79a495
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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()
   ), ...)