export check_args() (#1059)

Author: simonpcouch

NAMESPACE

@@ -5,6 +5,29 @@ S3method(.censoring_weights_graf,model_fit)
 S3method(augment,model_fit)
 S3method(autoplot,glmnet)
 S3method(autoplot,model_fit)
+S3method(check_args,C5_rules)
+S3method(check_args,bag_tree)
+S3method(check_args,boost_tree)
+S3method(check_args,cubist_rules)
+S3method(check_args,decision_tree)
+S3method(check_args,default)
+S3method(check_args,discrim_flexible)
+S3method(check_args,discrim_linear)
+S3method(check_args,discrim_regularized)
+S3method(check_args,linear_reg)
+S3method(check_args,logistic_reg)
+S3method(check_args,mars)
+S3method(check_args,mlp)
+S3method(check_args,multinom_reg)
+S3method(check_args,nearest_neighbor)
+S3method(check_args,pls)
+S3method(check_args,poisson_reg)
+S3method(check_args,rand_forest)
+S3method(check_args,surv_reg)
+S3method(check_args,survival_reg)
+S3method(check_args,svm_linear)
+S3method(check_args,svm_poly)
+S3method(check_args,svm_rbf)
 S3method(extract_fit_engine,model_fit)
 S3method(extract_parameter_dials,model_spec)
 S3method(extract_parameter_set_dials,model_spec)
@@ -180,6 +203,7 @@ export(bartMachine_interval_calc)
 export(boost_tree)
 export(case_weights_allowed)
 export(cforest_train)
+export(check_args)
 export(check_empty_ellipse)
 export(check_final_param)
 export(condense_control)

R/bag_tree.R

@@ -84,6 +84,7 @@ update.bag_tree <-
 
 # ------------------------------------------------------------------------------
 
+#' @export
 check_args.bag_tree <- function(object) {
   if (object$engine == "C5.0" && object$mode == "regression")
     stop("C5.0 is classification only.", call. = FALSE)

R/boost_tree.R

@@ -163,6 +163,7 @@ translate.boost_tree <- function(x, engine = x$engine, ...) {
 
 # ------------------------------------------------------------------------------
 
+#' @export
 check_args.boost_tree <- function(object) {
 
   args <- lapply(object$args, rlang::eval_tidy)

R/c5_rules.R

@@ -110,6 +110,7 @@ update.C5_rules <-
 
 # make work in different places
 
+#' @export
 check_args.C5_rules <- function(object) {
 
   args <- lapply(object$args, rlang::eval_tidy)

R/cubist_rules.R

@@ -134,6 +134,7 @@ update.cubist_rules <-
 
 # make work in different places
 
+#' @export
 check_args.cubist_rules <- function(object) {
 
   args <- lapply(object$args, rlang::eval_tidy)

R/decision_tree.R

@@ -127,6 +127,7 @@ translate.decision_tree <- function(x, engine = x$engine, ...) {
 
 # ------------------------------------------------------------------------------
 
+#' @export
 check_args.decision_tree <- function(object) {
   if (object$engine == "C5.0" && object$mode == "regression")
     rlang::abort("C5.0 is classification only.")

R/discrim_flexible.R

@@ -84,6 +84,7 @@ update.discrim_flexible <-
 
 # ------------------------------------------------------------------------------
 
+#' @export
 check_args.discrim_flexible <- function(object) {
 
   args <- lapply(object$args, rlang::eval_tidy)

R/discrim_linear.R

@@ -79,6 +79,7 @@ update.discrim_linear <-
 
 # ------------------------------------------------------------------------------
 
+#' @export
 check_args.discrim_linear <- function(object) {
 
   args <- lapply(object$args, rlang::eval_tidy)

R/discrim_regularized.R

@@ -94,6 +94,7 @@ update.discrim_regularized <-
 
 # ------------------------------------------------------------------------------
 
+#' @export
 check_args.discrim_regularized <- function(object) {
 
   args <- lapply(object$args, rlang::eval_tidy)

R/linear_reg.R

@@ -105,6 +105,7 @@ update.linear_reg <-
 
 # ------------------------------------------------------------------------------
 
+#' @export
 check_args.linear_reg <- function(object) {
 
   args <- lapply(object$args, rlang::eval_tidy)