prepare for hierarchical munging

syberia · Mar 18, 2015 · 9d1c7c4 · 9d1c7c4
1 parent 1bd2d1a
commit 9d1c7c4
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/R/tundra_container.r b/R/tundra_container.r
@@ -15,7 +15,7 @@ tundra_container <- setRefClass('tundraContainer',  #define reference classes to
   fields = list(keyword = 'character',
                 train_fn = 'function',
                 predict_fn = 'function',
-                munge_procedure = 'list',  # tundra contains munge_procedure so that it remembers the data-prep steps
+                munge_procedure = 'ANY',  # tundra contains munge_procedure so that it remembers the data-prep steps
                 default_args = 'list',
                 trained = 'logical',
                 input = 'list',
@@ -28,6 +28,9 @@ tundra_container <- setRefClass('tundraContainer',  #define reference classes to
                           munge_procedure = list(),
                           default_args = list(),
                           internal = list()) {
+      if (!(is.list(munge_procedure) || is(munge_procedure, "stageRunner"))) {
+        stop("munge_procedure must be a list or stageRunner", call. = FALSE)
+      }
       keyword <<- keyword
       train_fn <<- train_fn
       predict_fn <<- predict_fn