Fixed a bug induced by issue #537

topepo · Jan 9, 2017 · 1d6f442 · 1d6f442
1 parent 854d128
commit 1d6f442
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pkg/caret/R/train.default.R b/pkg/caret/R/train.default.R
@@ -445,7 +445,7 @@ train.default <- function(x, y,
     } else {
       tuneGrid <- models$grid(x = x, y = y, len = tuneLength, search = trControl$search)
       if (tuneLength < nrow(tuneGrid))
-        tuneGrid <- tuneGrid[1:tuneLength, ]
+        tuneGrid <- tuneGrid[1:tuneLength,,drop = FALSE]
     }
   }