Use rlang arg_match() to prevent partial match

epiforecasts · Nov 1, 2023 · d5c3ebb · d5c3ebb
1 parent c277d5c
commit d5c3ebb
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 6 deletions.
diff --git a/NAMESPACE b/NAMESPACE
@@ -198,6 +198,7 @@ importFrom(purrr,safely)
 importFrom(purrr,transpose)
 importFrom(purrr,walk)
 importFrom(rlang,abort)
+importFrom(rlang,arg_match)
 importFrom(rlang,cnd_muffle)
 importFrom(rlang,warn)
 importFrom(rstan,expose_stan_functions)

diff --git a/R/checks.R b/R/checks.R
@@ -24,13 +24,14 @@ check_reports_valid <- function(reports, model) {
   # Check that the case time series (reports) is a data frame
   assert_data_frame(reports)
   # Perform checks depending on the model to the data is meant to be used with
-  model <- match.arg(
-    model,
-    c("estimate_infections",
-      "estimate_truncation",
-      "estimate_secondary"
-      )
+  model <- arg_match(
+  model,
+  values = c(
+    "estimate_infections",
+    "estimate_truncation",
+    "estimate_secondary"
     )
+  )
 
   if (model == "estimate_secondary") {
     # Check that reports has the right column names