diff --git a/bilby/core/sampler/pymc3.py b/bilby/core/sampler/pymc3.py index a6e83cc936db564f1baccc0950d025b0d4239967..7b9d6e675d40b1f05ac4b70bcc17e7957a4ec4e5 100644 --- a/bilby/core/sampler/pymc3.py +++ b/bilby/core/sampler/pymc3.py @@ -57,8 +57,8 @@ class Pymc3(MCMCSampler): default_kwargs = dict( draws=500, step=None, init='auto', n_init=200000, start=None, trace=None, chain_idx=0, chains=2, cores=1, tune=500, nuts_kwargs=None, step_kwargs=None, progressbar=True, - model=None, random_seed=None, live_plot=False, discard_tuned_samples=True, - live_plot_kwargs=None, compute_convergence_checks=True) + model=None, random_seed=None, discard_tuned_samples=True, + compute_convergence_checks=True) def __init__(self, likelihood, priors, outdir='outdir', label='label', use_ratio=False, plot=False, diff --git a/test/sampler_test.py b/test/sampler_test.py index 8f58b70dca460ac46949e2d810b8f2847dc7719a..8053ff8c54164dde94a18b20829e9947b0a4c81a 100644 --- a/test/sampler_test.py +++ b/test/sampler_test.py @@ -370,16 +370,16 @@ class TestPyMC3(unittest.TestCase): expected = dict( draws=500, step=None, init='auto', n_init=200000, start=None, trace=None, chain_idx=0, chains=2, cores=1, tune=500, nuts_kwargs=None, step_kwargs=None, progressbar=True, - model=None, random_seed=None, live_plot=False, discard_tuned_samples=True, - live_plot_kwargs=None, compute_convergence_checks=True) + model=None, random_seed=None, discard_tuned_samples=True, + compute_convergence_checks=True) self.assertDictEqual(expected, self.sampler.kwargs) def test_translate_kwargs(self): expected = dict( draws=500, step=None, init='auto', n_init=200000, start=None, trace=None, chain_idx=0, chains=2, cores=1, tune=500, nuts_kwargs=None, step_kwargs=None, progressbar=True, - model=None, random_seed=None, live_plot=False, discard_tuned_samples=True, - live_plot_kwargs=None, compute_convergence_checks=True) + model=None, random_seed=None, discard_tuned_samples=True, + compute_convergence_checks=True) self.sampler.kwargs['draws'] = 123 for equiv in bilby.core.sampler.base_sampler.NestedSampler.npoints_equiv_kwargs: new_kwargs = self.sampler.kwargs.copy()