Merge pull request #3296 from stan-dev/fix/no-adapt-sampler-output-fo…

…rmatting Ensure formatting of output is same for adapt=false samplers
stan-dev · Aug 8, 2024 · 4c5c89e · 4c5c89e
2 parents eeaa211 + 1d13747
commit 4c5c89e
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/src/stan/services/sample/hmc_nuts_dense_e.hpp b/src/stan/services/sample/hmc_nuts_dense_e.hpp
@@ -237,7 +237,8 @@ int hmc_nuts_dense_e(Model& model, size_t num_chains,
             util::run_sampler(samplers[i], model, cont_vectors[i], num_warmup,
                               num_samples, num_thin, refresh, save_warmup,
                               rngs[i], interrupt, logger, sample_writer[i],
-                              diagnostic_writer[i], init_chain_id + i);
+                              diagnostic_writer[i], init_chain_id + i,
+                              num_chains);
           }
         },
         tbb::simple_partitioner());

diff --git a/src/stan/services/sample/hmc_nuts_diag_e.hpp b/src/stan/services/sample/hmc_nuts_diag_e.hpp
@@ -232,7 +232,8 @@ int hmc_nuts_diag_e(Model& model, size_t num_chains,
           util::run_sampler(samplers[i], model, cont_vectors[i], num_warmup,
                             num_samples, num_thin, refresh, save_warmup,
                             rngs[i], interrupt, logger, sample_writer[i],
-                            diagnostic_writer[i], init_chain_id + i);
+                            diagnostic_writer[i], init_chain_id + i,
+                            num_chains);
         }
       },
       tbb::simple_partitioner());