Addressing issues

UofUEpiBio · Sep 27, 2024 · 58bfc48 · 58bfc48
1 parent 5528890
commit 58bfc48
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/R/tool.R b/R/tool.R
@@ -191,12 +191,12 @@ add_tool <- function(model, tool, proportion) {
       distfun = distribute_tool_randomly(proportion, TRUE)
     )
 
-  } else if (attr(tool, "uses_deprecated")) {
+  } else if (isTRUE(attr(tool, "uses_deprecated"))) {
 
     set_distribution_tool(
       tool = tool,
       distfun = distribute_tool_randomly(
-        proportion = attr(tool, "deprecated_args")$prevalence,
+        prevalence = attr(tool, "deprecated_args")$prevalence,
         as_proportion = attr(tool, "deprecated_args")$as_proportion
       )
     )

diff --git a/R/virus.R b/R/virus.R
@@ -191,12 +191,12 @@ add_virus <- function(model, virus, proportion) {
       distfun=distribute_virus_randomly(proportion, as_proportion = TRUE)
     )
 
-  } else if (attr(tool, "uses_deprecated")) {
+  } else if (isTRUE(attr(tool, "uses_deprecated"))) {
 
     set_distribution_virus(
       virus = tool,
       distfun = distribute_virus_randomly(
-        proportion = attr(tool, "deprecated_args")$prevalence,
+        prevalence = attr(tool, "deprecated_args")$prevalence,
         as_proportion = attr(tool, "deprecated_args")$as_proportion
       )
     )