From d36b447a0902801811f4be51b06219536c176b6d Mon Sep 17 00:00:00 2001
From: Hugo Gruson <10783929+Bisaloo@users.noreply.github.com>
Date: Thu, 22 Feb 2024 15:14:15 +0100
Subject: [PATCH] Get i2extras to work with SI passed as epidist

---
 .../templates/transmissibility/skeleton/rmdchunks/i2extras.Rmd  | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/inst/rmarkdown/templates/transmissibility/skeleton/rmdchunks/i2extras.Rmd b/inst/rmarkdown/templates/transmissibility/skeleton/rmdchunks/i2extras.Rmd
index 44cd846e..b03fdc07 100644
--- a/inst/rmarkdown/templates/transmissibility/skeleton/rmdchunks/i2extras.Rmd
+++ b/inst/rmarkdown/templates/transmissibility/skeleton/rmdchunks/i2extras.Rmd
@@ -141,7 +141,7 @@ This section contains:
 
 ```{r }
 res_R_wl <- last_trends %>%
-  mutate(R = map(model, epitrix::lm2R0_sample, w = si, 500)) %>%
+  mutate(R = map(model, epitrix::lm2R0_sample, w = si$prob_dist$d(si_x))) %>%
   dplyr::select({{ group_var }}, R) %>%
   unnest(R)