fix problems with chf

ModelOriented · Oct 25, 2023 · 8c0c8fd · 8c0c8fd
1 parent 460f253
commit 8c0c8fd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/R/unify_ranger_surv.R b/R/unify_ranger_surv.R
@@ -82,7 +82,7 @@ ranger_surv.unify <- function(rf_model, data, type = c("risk", "survival", "chf"
   stopifnot(
     "`times` must be a numeric vector and argument \
     `type = 'survival'` or `type = 'chf'` must be set." =
-      ifelse(!is.null(times), is.numeric(times) && type == "survival", TRUE)
+      ifelse(!is.null(times), is.numeric(times) && (type == "survival" || type == "chf"), TRUE)
   )
 
   surv_common <- ranger_surv.common(rf_model, data)