Skip to content

Commit 2ad7ec3

Browse files
committed
Make stop_incompatible_* functions with one arg
1 parent a2ec3f9 commit 2ad7ec3

File tree

3 files changed

+12
-10
lines changed

3 files changed

+12
-10
lines changed

R/aaa_models.R

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -134,20 +134,21 @@ check_mode_val <- function(mode) {
134134
}
135135

136136

137-
glue_compatible_modes <- function(cls, spec_modes) {
138-
glue::glue(
137+
stop_incompatible_mode <- function(spec_modes) {
138+
msg <- glue::glue(
139139
"Available modes are: ",
140140
glue::glue_collapse(glue::glue("'{spec_modes}'"), sep = ", ")
141141
)
142+
rlang::abort(msg)
142143
}
143144

144145
# check if class and mode are compatible
145146
check_spec_mode_val <- function(cls, mode) {
146147
spec_modes <- rlang::env_get(get_model_env(), paste0(cls, "_modes"))
147148
if (is.null(mode) || length(mode) > 1) {
148-
rlang::abort(glue_compatible_modes(cls, spec_modes))
149+
stop_incompatible_mode(spec_modes)
149150
} else if (!(mode %in% spec_modes)) {
150-
rlang::abort(glue_compatible_modes(cls, spec_modes))
151+
stop_incompatible_mode(spec_modes)
151152
}
152153
invisible(NULL)
153154
}

R/arguments.R

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -79,7 +79,7 @@ set_mode <- function(object, mode) {
7979
cls <- class(object)[1]
8080
if (rlang::is_missing(mode)) {
8181
spec_modes <- rlang::env_get(get_model_env(), paste0(cls, "_modes"))
82-
rlang::abort(glue_compatible_modes(cls, spec_modes))
82+
stop_incompatible_mode(spec_modes)
8383
}
8484
check_spec_mode_val(cls, mode)
8585
object$mode <- mode

R/engines.R

Lines changed: 6 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -10,20 +10,21 @@ possible_engines <- function(object, ...) {
1010
unique(engs$engine)
1111
}
1212

13-
glue_compatible_engines <- function(engine, avail_eng) {
14-
glue::glue(
13+
stop_incompatible_engine <- function(avail_eng) {
14+
msg <- glue::glue(
1515
"Available engines are: ",
1616
glue::glue_collapse(glue::glue("'{avail_eng}'"), sep = ", ")
1717
)
18+
rlang::abort(msg)
1819
}
1920

2021
check_engine <- function(object) {
2122
avail_eng <- possible_engines(object)
2223
eng <- object$engine
2324
if (is.null(eng) || length(eng) > 1) {
24-
rlang::abort(glue_compatible_engines(eng, avail_eng))
25+
stop_incompatible_engine(avail_eng)
2526
} else if (!(eng %in% avail_eng)) {
26-
rlang::abort(glue_compatible_engines(eng, avail_eng))
27+
stop_incompatible_engine(avail_eng)
2728
}
2829
object
2930
}
@@ -95,7 +96,7 @@ set_engine <- function(object, engine, ...) {
9596

9697
if (rlang::is_missing(engine)) {
9798
avail_eng <- possible_engines(object)
98-
rlang::abort(glue_compatible_engines(object$engine, avail_eng))
99+
stop_incompatible_engine(avail_eng)
99100
}
100101
object$engine <- engine
101102
object <- check_engine(object)

0 commit comments

Comments
 (0)