diff --git a/examples/core_examples/alternative_samplers/linear_regression_pymc_custom_likelihood.py b/examples/core_examples/alternative_samplers/linear_regression_pymc_custom_likelihood.py index e9763770c8ad96b32adbf256cc44aabf0ceffb1e..ae6c58a27dad02645813d5e4bf3b16ecd20cf4b4 100644 --- a/examples/core_examples/alternative_samplers/linear_regression_pymc_custom_likelihood.py +++ b/examples/core_examples/alternative_samplers/linear_regression_pymc_custom_likelihood.py @@ -81,7 +81,7 @@ class GaussianLikelihoodPyMC(bilby.core.likelihood.GaussianLikelihood): likelihood is evaluated. """ - from bilby.core.sampler import Pymc + from bilby.core.sampler.pymc import Pymc if not isinstance(sampler, Pymc): return super(GaussianLikelihoodPyMC, self).log_likelihood()