From 9d7f5b3a9e89f9c3d46b70404b0fe10b75eb9fed Mon Sep 17 00:00:00 2001 From: turnmanh <17703667+turnmanh@users.noreply.github.com> Date: Wed, 19 Feb 2025 19:56:14 +0100 Subject: [PATCH] duplication of 'large number' in warning --- sbi/inference/posteriors/mcmc_posterior.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/sbi/inference/posteriors/mcmc_posterior.py b/sbi/inference/posteriors/mcmc_posterior.py index 674aefb04..e2509a359 100644 --- a/sbi/inference/posteriors/mcmc_posterior.py +++ b/sbi/inference/posteriors/mcmc_posterior.py @@ -452,11 +452,11 @@ def sample_batched( num_chains_extended = batch_size * num_chains if num_chains_extended > 100: warnings.warn( - "Note that for batched sampling, we use num_chains many chains for each" - " x in the batch. With the given settings, this results in a large " - f"number large number of chains ({num_chains_extended}), which can be " - "slow and memory-intensive for vectorized MCMC. Consider reducing the " - "number of chains.", + "Note that for batched sampling, we use num_chains many chains " + "for each x in the batch. With the given settings, this results " + f"in a large number of chains ({num_chains_extended}), which can " + "be slow and memory-intensive for vectorized MCMC. Consider " + "reducing the number of chains or batch size.", stacklevel=2, ) init_strategy_parameters["num_return_samples"] = num_chains_extended