emcee: vectorize sampling since all ln_pdfs accept numpy arrays

GeoStat-Framework · Apr 17, 2024 · c88f7ba · c88f7ba
1 parent 228edf9
commit c88f7ba
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/gstools/random/rng.py b/src/gstools/random/rng.py
@@ -86,7 +86,7 @@ def sample_ln_pdf(
             self.random.rand(nwalkers).reshape((nwalkers, 1)) * sample_around
         )
         # initialize the sampler
-        sampler = mc.EnsembleSampler(nwalkers, 1, ln_pdf)
+        sampler = mc.EnsembleSampler(nwalkers, 1, ln_pdf, vectorize=True)
         # burn in phase with saving of last position
         initial_state = State(init_guess, copy=True)
         initial_state.random_state = self.random.get_state()