DS-1526 update rPartToTreeFrame

Displayr · Sep 18, 2017 · f51c63c · f51c63c
1 parent 47c0920
commit f51c63c
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/R/rparttree.R b/R/rparttree.R
@@ -41,7 +41,10 @@ rPartToTreeFrame <- function(obj)
     }
     frame$splits <- splits
 
-    outcome.var <- obj$model[[OutcomeName(obj$terms)]]
+    formula <- obj$terms
+    # remove "terms" class which confuses formula.tools used by OutcomeName
+    class(formula) <- "formula"
+    outcome.var <- obj$model[[OutcomeName(formula)]]
 
     if (is.factor(outcome.var))
         frame$yval <- levels(outcome.var)[frame$yval]