Add chain id

mrc-ide · Nov 4, 2024 · 6520b77 · 6520b77
1 parent af3ff83
commit 6520b77
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/R/runner-simultaneous.R b/R/runner-simultaneous.R
@@ -128,6 +128,8 @@ monty_run_chains_simultaneous2 <- function(chain_state, model, sampler,
   history_pars <- array(NA_real_, c(n_pars, n_steps_record, n_chains))
   history_density <- matrix(NA_real_, n_steps_record, n_chains)
 
+  chain_id <- seq_len(n_chains)
+
   for (i in seq_len(steps$total)) {
     chain_state <- sampler$step(chain_state, model, rng)
     history_pars[, i, ] <- chain_state$pars