export methods for censoring model (#966)

Author: hfrick

DESCRIPTION

@@ -1,6 +1,6 @@
 Package: parsnip
 Title: A Common API to Modeling and Analysis Functions
-Version: 1.1.0.9001
+Version: 1.1.0.9002
 Authors@R: c(
     person("Max", "Kuhn", , "[email protected]", role = c("aut", "cre")),
     person("Davis", "Vaughan", , "[email protected]", role = "aut"),

NAMESPACE

@@ -35,6 +35,7 @@ S3method(predict,"_elnet")
 S3method(predict,"_glmnetfit")
 S3method(predict,"_lognet")
 S3method(predict,"_multnet")
+S3method(predict,censoring_model)
 S3method(predict,censoring_model_reverse_km)
 S3method(predict,model_fit)
 S3method(predict,model_spec)
@@ -61,6 +62,7 @@ S3method(predict_raw,"_multnet")
 S3method(predict_raw,model_fit)
 S3method(predict_survival,model_fit)
 S3method(predict_time,model_fit)
+S3method(print,censoring_model)
 S3method(print,control_parsnip)
 S3method(print,model_fit)
 S3method(print,model_spec)

R/survival-censoring-model.R

@@ -46,11 +46,13 @@ reverse_km <- function(obj, eval_env) {
 # ------------------------------------------------------------------------------
 # Basic S3 methods
 
+#' @export
 print.censoring_model <- function(x, ...) {
   cat(x$label, "model for predicting the probability of censoring\n")
   invisible(x)
 }
 
+#' @export
 predict.censoring_model <- function(object, ...) {
   rlang::abort(
     paste("Don't know how to predict with a censoring model of type:", object$label)