diff --git a/DESCRIPTION b/DESCRIPTION index e45ba627..d7f344a3 100644 --- a/DESCRIPTION +++ b/DESCRIPTION @@ -1,7 +1,7 @@ Type: Package Package: yardstick Title: Tidy Characterizations of Model Performance -Version: 1.2.0.9002 +Version: 1.2.0.9003 Authors@R: c( person("Max", "Kuhn", , "max@posit.co", role = "aut"), person("Davis", "Vaughan", , "davis@posit.co", role = "aut"), diff --git a/R/surv-brier_survival.R b/R/surv-brier_survival.R index e9f172cd..8903fee0 100644 --- a/R/surv-brier_survival.R +++ b/R/surv-brier_survival.R @@ -138,6 +138,7 @@ brier_survival_impl <- function(truth, surv_status <- .extract_surv_status(truth) if (!is.null(case_weights)) { + case_weights <- vec_cast(case_weights, to = double()) norm_const <- sum(case_weights) } else { case_weights <- rep(1, length(estimate)) diff --git a/R/surv-concordance_survival.R b/R/surv-concordance_survival.R index 506ab607..25836fd7 100644 --- a/R/surv-concordance_survival.R +++ b/R/surv-concordance_survival.R @@ -104,6 +104,8 @@ concordance_survival_vec <- function(truth, concordance_survival_impl <- function(truth, estimate, case_weights) { if (is.null(case_weights)) { case_weights <- rep(1, length(estimate)) + } else { + case_weights <- vec_cast(case_weights, to = double()) } survival::concordance(truth ~ estimate, weights = case_weights)$concordance