diff --git a/DESCRIPTION b/DESCRIPTION index 74d371b82..a1b30100b 100644 --- a/DESCRIPTION +++ b/DESCRIPTION @@ -1,6 +1,6 @@ Package: parsnip Title: A Common API to Modeling and Analysis Functions -Version: 1.1.1.9002 +Version: 1.1.1.9003 Authors@R: c( person("Max", "Kuhn", , "max@posit.co", role = c("aut", "cre")), person("Davis", "Vaughan", , "davis@posit.co", role = "aut"), diff --git a/R/standalone-survival.R b/R/standalone-survival.R index a75464e85..df4808081 100644 --- a/R/standalone-survival.R +++ b/R/standalone-survival.R @@ -74,11 +74,12 @@ .extract_surv_time <- function(surv) { .is_surv(surv) keepers <- c("time", "start", "stop", "time1", "time2") - res <- surv[, colnames(surv) %in% keepers] - if (NCOL(res) > 1) { + cols <- colnames(surv)[colnames(surv) %in% keepers] + res <- surv[, cols, drop = FALSE] + if (length(cols) > 1) { res <- tibble::tibble(as.data.frame(res)) } else { - res <- unname(res) + res <- as.numeric(res) } res }