mcmc_posterior imports shape handling from new location

sbi-dev · Aug 13, 2024 · 80e87b8 · 80e87b8
1 parent b0bd362
commit 80e87b8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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,