fixed control object printing

Author: topepo

NAMESPACE

@@ -32,6 +32,7 @@ S3method(predict_raw,"_multnet")
 S3method(predict_raw,model_fit)
 S3method(print,boost_tree)
 S3method(print,decision_tree)
+S3method(print,fit_control)
 S3method(print,linear_reg)
 S3method(print,logistic_reg)
 S3method(print,mars)

NEWS.md

@@ -11,6 +11,8 @@
 of possible varying arguments is returned (as opposed to only the arguments
 that are actually varying).
 
+* `fit_control()` not returns an S3 method. 
+
 ## Bug Fixes
 
 * `varying_args()` now uses the version from the `generics` package. This means

R/fit_control.R

@@ -26,12 +26,12 @@ fit_control <- function(verbosity = 1L, catch = FALSE) {
   res
 }
 
-# '@export
+#' @export
 print.fit_control <- function(x, ...) {
   cat("parsnip control object\n")
   if (x$verbosity > 1)
-    cat(" verbose mode\n")
-  if (x$catch > 1)
-    cat(" fit errors will be caught\n")
+    cat(" - verbose level", x$verbosity, "\n")
+  if (x$catch)
+    cat(" - fit errors will be caught\n")
   invisible(x)
 }