diff --git a/sbi/inference/posteriors/mcmc_posterior.py b/sbi/inference/posteriors/mcmc_posterior.py index 64327d3aa..6180041cd 100644 --- a/sbi/inference/posteriors/mcmc_posterior.py +++ b/sbi/inference/posteriors/mcmc_posterior.py @@ -21,7 +21,7 @@ from sbi.inference.posteriors.base_posterior import NeuralPotentialPosterior from sbi.inference.potentials.base_potential import BasePotential -from sbi.neural_nets.density_estimators.shape_handling import reshape_to_batch_event +from sbi.neural_nets.estimators.shape_handling import reshape_to_batch_event from sbi.samplers.mcmc import ( IterateParameters, PyMCSampler,