Skip to content

Commit 48efe84

Browse files
authored
speed up get_args() helper (#933)
1 parent 2f886a3 commit 48efe84

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

R/translate.R

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -127,9 +127,12 @@ get_model_spec <- function(model, mode, engine) {
127127

128128
get_args <- function(model, engine) {
129129
m_env <- get_model_env()
130-
rlang::env_get(m_env, paste0(model, "_args")) %>%
131-
dplyr::filter(engine == !!engine) %>%
132-
dplyr::select(-engine)
130+
131+
args <- rlang::env_get(m_env, paste0(model, "_args"))
132+
args <- vctrs::vec_slice(args, args$engine == engine)
133+
args$engine <- NULL
134+
135+
args
133136
}
134137

135138
# to replace harmonize

0 commit comments

Comments
 (0)