recode: restructured modNLSCollect as list

ryomayes · Aug 12, 2022 · ef12599 · ef12599
1 parent 1935868
commit ef12599
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 8 deletions.
diff --git a/R/R/inputs.R b/R/R/inputs.R
@@ -721,12 +721,11 @@ robyn_engineering <- function(x, quiet = FALSE, ...) {
   InputCollect[["dt_mod"]] <- dt_transform
   InputCollect[["dt_modRollWind"]] <- dt_transform[rollingWindowStartWhich:rollingWindowEndWhich, ]
   InputCollect[["dt_inputRollWind"]] <- dt_inputRollWind
-  InputCollect[["modNLSCollect"]] <- modNLSCollect
-  InputCollect[["plotNLSCollect"]] <- plotNLSCollect
-  InputCollect[["yhatNLSCollect"]] <- yhatNLSCollect
-
-  # InputCollect[["exposure_selector"]] <- exposure_selector
-  # InputCollect[["mediaCostFactor"]] <- mediaCostFactor
+  InputCollect[["modNLS"]] <- list(
+    results = modNLSCollect,
+    yhat = yhatNLSCollect,
+    plots = plotNLSCollect
+  )
   return(InputCollect)
 }
 

diff --git a/demo/demo.R b/demo/demo.R
@@ -271,8 +271,8 @@ print(InputCollect)
 
 #### Check spend exposure fit if available
 if (length(InputCollect$exposure_vars) > 0) {
-  InputCollect$plotNLSCollect$facebook_I
-  InputCollect$plotNLSCollect$search_clicks_P
+  InputCollect$modNLS$plots$facebook_I
+  InputCollect$modNLS$plots$search_clicks_P
 }
 
 ##### Manually save and import InputCollect as JSON file