diff --git a/R/shaft-.R b/R/shaft-.R index 1f4a6f5d6..d4d8de423 100644 --- a/R/shaft-.R +++ b/R/shaft-.R @@ -168,6 +168,16 @@ pillar_shaft.Surv2 <- function(x, ...) { new_pillar_shaft_simple(format(x), align = "right") } +# registered in .onLoad() +type_sum.Surv <- function(x) { + "Surv" +} + +# registered in .onLoad() +type_sum.Surv2 <- function(x) { + "Surv2" +} + #' @export #' @rdname pillar_shaft pillar_shaft.Date <- function(x, ...) { diff --git a/R/zzz.R b/R/zzz.R index fcfd46131..c105c5092 100644 --- a/R/zzz.R +++ b/R/zzz.R @@ -21,7 +21,9 @@ NULL register_s3_method("vctrs", "vec_ptype_abbr", "pillar_empty_col") register_s3_method("bit64", "pillar_shaft", "integer64", gen_pkg = "pillar") register_s3_method("survival", "pillar_shaft", "Surv", gen_pkg = "pillar") + register_s3_method("survival", "type_sum", "Surv", gen_pkg = "pillar") register_s3_method("survival", "pillar_shaft", "Surv2", gen_pkg = "pillar") + register_s3_method("survival", "type_sum", "Surv2", gen_pkg = "pillar") assign_crayon_styles()