Skip to content
Snippets Groups Projects
Commit bb27651c authored by Matthew Pitkin's avatar Matthew Pitkin
Browse files

Fix sampler test failure

parent 8ce9b4e4
No related branches found
No related tags found
1 merge request!814Fix PyMC3 after deprecation of step_kwargs and nuts_kwargs
......@@ -703,8 +703,8 @@ class TestPyMC3(unittest.TestCase):
discard_tuned_samples=True,
compute_convergence_checks=True,
)
expected.update(bilby.core.sampler.Pymc3.default_nuts_kwargs)
expected.update(bilby.core.sampler.Pymc3.default_step_kwargs)
expected.update(self.sampler.default_nuts_kwargs)
expected.update(self.sampler.default_step_kwargs)
self.assertDictEqual(expected, self.sampler.kwargs)
def test_translate_kwargs(self):
......@@ -727,8 +727,8 @@ class TestPyMC3(unittest.TestCase):
discard_tuned_samples=True,
compute_convergence_checks=True,
)
expected.update(bilby.core.sampler.Pymc3.default_nuts_kwargs)
expected.update(bilby.core.sampler.Pymc3.default_step_kwargs)
expected.update(self.sampler.default_nuts_kwargs)
expected.update(self.sampler.default_step_kwargs)
self.sampler.kwargs["draws"] = 123
for equiv in bilby.core.sampler.base_sampler.NestedSampler.npoints_equiv_kwargs:
new_kwargs = self.sampler.kwargs.copy()
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment