fixed bug in code that gets dependencies

Author: topepo

R/surv_reg_data.R

@@ -4,9 +4,9 @@ set_model_mode("surv_reg", "regression")
 
 # ------------------------------------------------------------------------------
 
-set_model_engine("surv_reg", "regression", "flexsurv")
-set_dependency("surv_reg", "flexsurv", "flexsurv")
-set_dependency("surv_reg", "flexsurv", "survival")
+set_model_engine("surv_reg", mode = "regression", eng = "flexsurv")
+set_dependency("surv_reg", eng = "flexsurv", pkg = "flexsurv")
+set_dependency("surv_reg", eng = "flexsurv", pkg = "survival")
 
 set_model_arg(
   model = "surv_reg",
@@ -68,8 +68,8 @@ set_pred(
 
 # ------------------------------------------------------------------------------
 
-set_model_engine("surv_reg", "regression", "survival")
-set_dependency("surv_reg", "survival", "survival")
+set_model_engine("surv_reg", mode = "regression", eng = "survival")
+set_dependency("surv_reg", eng = "survival", pkg = "survival")
 
 set_model_arg(
   model = "surv_reg",

R/translate.R

@@ -127,6 +127,7 @@ get_model_spec <- function(model, mode, engine) {
   res <- list()
   res$libs <-
     rlang::env_get(m_env, paste0(model, "_pkgs")) %>%
+    dplyr::filter(engine == !!engine) %>%
     purrr::pluck("pkg") %>%
     purrr::pluck(1)