Commit 0f442275 authored by Leo Pound Singer's avatar Leo Pound Singer

Pass through kwargs to sampler

parent 8d654e37
...@@ -99,8 +99,8 @@ def ez_emcee(log_prob_fn, lo, hi, nindep=200, ...@@ -99,8 +99,8 @@ def ez_emcee(log_prob_fn, lo, hi, nindep=200,
with tqdm(total=nburnin + nindep * nsteps) as progress: with tqdm(total=nburnin + nindep * nsteps) as progress:
sampler = Sampler(nwalkers, ndim, log_prob_fn, logp, sampler = Sampler(nwalkers, ndim, log_prob_fn, logp,
ntemps=ntemps, loglargs=args, logpargs=[lo, hi], ntemps=ntemps, loglargs=args, loglkwargs=kwargs,
**options) logpargs=[lo, hi], **options)
pos = np.random.uniform(lo, hi, (ntemps, nwalkers, ndim)) pos = np.random.uniform(lo, hi, (ntemps, nwalkers, ndim))
# Burn in # Burn in
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment