fix strata selection in print outcomes

HopkinsIDD · Apr 17, 2024 · c3a6184 · c3a6184
1 parent 2ce74d0
commit c3a6184
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/flepimop/R_packages/flepiconfig/R/yaml_utils.R b/flepimop/R_packages/flepiconfig/R/yaml_utils.R
@@ -1335,9 +1335,12 @@ print_outcomes <- function (resume_modifier = NULL,
   incidC_pert <- ""
   pert_repeat <- prod(compartment_tract_length[-1], na.rm = TRUE)
 
-  variant_strata = compartment_tract_list$variant_strata
-  vaccine_strata = compartment_tract_list$vaccine_strata
-  age_strata = compartment_tract_list$age_strata
+  variant_strata = `$`(compartment_tract_list, 'variant')
+  vaccine_strata = `$`(compartment_tract_list, 'vacc')
+  age_strata = `$`(compartment_tract_list, 'age')
+#   variant_strata = compartment_tract_list$variant_strata
+#   vaccine_strata = compartment_tract_list$vaccine_strata
+#   age_strata = compartment_tract_list$age_strata
 
   for (i in 1:pert_repeat) {
     if (incidC_perturbation) {