use assert_class instead of is_epidist for input checking

epiverse-trace · Dec 6, 2023 · b3efd03 · b3efd03
1 parent 79f9ae9
commit b3efd03
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git a/R/probability_contain.R b/R/probability_contain.R
@@ -37,7 +37,7 @@ probability_contain <- function(R, k, num_init_infect = 1, control,
   }
   # check inputs
   if (input_params) {
-    epiparameter::is_epidist(offspring_dist)
+    checkmate::assert_class(offspring_dist, classes = "epidist")
     R <- get_epidist_param(epidist = offspring_dist, parameter = "R")
     k <- get_epidist_param(epidist = offspring_dist, parameter = "k")
   }

diff --git a/R/probability_epidemic.R b/R/probability_epidemic.R
@@ -36,7 +36,7 @@ probability_epidemic <- function(R, k, num_init_infect, ..., offspring_dist) {
   # check inputs
   chkDots(...)
   if (input_params) {
-    epiparameter::is_epidist(offspring_dist)
+    checkmate::assert_class(offspring_dist, classes = "epidist")
     R <- get_epidist_param(epidist = offspring_dist, parameter = "R")
     k <- get_epidist_param(epidist = offspring_dist, parameter = "k")
   }

diff --git a/R/proportion_cluster_size.R b/R/proportion_cluster_size.R
@@ -46,7 +46,7 @@ proportion_cluster_size <- function(R, k, cluster_size, ..., offspring_dist) {
   # check inputs
   chkDots(...)
   if (input_params) {
-    epiparameter::is_epidist(offspring_dist)
+    checkmate::assert_class(offspring_dist, classes = "epidist")
     R <- get_epidist_param(epidist = offspring_dist, parameter = "R")
     k <- get_epidist_param(epidist = offspring_dist, parameter = "k")
   }

diff --git a/R/proportion_transmission.R b/R/proportion_transmission.R
@@ -74,7 +74,7 @@ proportion_transmission <- function(R, k,
   # check inputs
   chkDots(...)
   if (input_params) {
-    epiparameter::is_epidist(offspring_dist)
+    checkmate::assert_class(offspring_dist, classes = "epidist")
     R <- get_epidist_param(epidist = offspring_dist, parameter = "R")
     k <- get_epidist_param(epidist = offspring_dist, parameter = "k")
   }