feat: include refreshChain on InputCollect

ryomayes · Aug 10, 2022 · e1c4399 · e1c4399
1 parent 45dcc19
commit e1c4399
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/R/R/json.R b/R/R/json.R
@@ -258,5 +258,6 @@ robyn_chain <- function(json_file) {
   dirs <- sapply(chainData, function(x) x$ExportedModel$plot_folder)
   json_files <- paste0(dirs, "RobynModel-", names(dirs), ".json")
   attr(chainData, "json_files") <- json_files
+  attr(chainData, "chain") <- names(chainData)
   return(invisible(chainData))
 }
diff --git a/R/R/refresh.R b/R/R/refresh.R
@@ -126,6 +126,8 @@ robyn_refresh <- function(json_file = NULL,
         quiet = FALSE, ...)
       listInit$InputCollect$refreshDepth <- str_count(json$ExportedModel$plot_folder, "_rf/") + 1
       listInit$InputCollect$refreshSourceID <- json$ExportedModel$select_model
+      chainData <- robyn_chain(json_file)
+      listInit$InputCollect$refreshChain <- attr(chainData, "chain")
       listInit$OutputCollect$hyper_updated <- json$ExportedModel$hyper_updated
       Robyn[["listInit"]] <- listInit
       objectPath <- json$ExportedModel$plot_folder