check if the list entry is of length 1, otherwise can unlist it in 2.15

viddtvcom · Jun 27, 2014 · 616dc08 · 616dc08
1 parent a0dfd1f
commit 616dc08
Showing 1 changed file with 12 additions and 2 deletions.
diff --git a/R/h2o-package/R/Internal.R b/R/h2o-package/R/Internal.R
@@ -416,7 +416,12 @@ function(expr) {
 function(some_expr_list) {
   len <- length(some_expr_list)
   while(len > 1) {
-    num_sub_lists <- length(unlist(some_expr_list[[len]])) / length(some_expr_list[[len]])
+    num_sub_lists <- 0
+    if (length(some_expr_list[[len]]) == 1) {
+      num_sub_lists <- 1
+    } else {
+      num_sub_lists <- length(unlist(some_expr_list[[len]])) / length(some_expr_list[[len]])
+    }
     if (num_sub_lists > 1) {
       some_expr_list[[len]] <- .back_to_expr(some_expr_list[[len]])
     } else if (is.atomic(some_expr_list[[len]]) || is.name(some_expr_list[[len]])) {
@@ -511,7 +516,12 @@ function(some_expr_list, envir) {
   while(i <= len) {
 
     # Check if there are sub lists and recurse them
-    num_sub_lists <- length(unlist(some_expr_list[[i]])) / length(some_expr_list[[i]])
+    num_sub_lists <- 0
+    if (length(some_expr_list[[i]]) == 1) {
+      num_sub_lists <- 1
+    } else {
+      num_sub_lists <- length(unlist(some_expr_list[[i]])) / length(some_expr_list[[i]])
+    }
     if (num_sub_lists > 1) {
 
       # recurse on the sublist