From ce42badc26c02cc9130c24c90f598422681e1fd8 Mon Sep 17 00:00:00 2001 From: michaeldeistler Date: Tue, 23 Aug 2022 18:39:33 +0200 Subject: [PATCH] allow more flexible proposals in atomic snpe-c --- sbi/inference/snpe/snpe_c.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/sbi/inference/snpe/snpe_c.py b/sbi/inference/snpe/snpe_c.py index 34c53ab4b..65d3ac04f 100644 --- a/sbi/inference/snpe/snpe_c.py +++ b/sbi/inference/snpe/snpe_c.py @@ -161,7 +161,8 @@ def train( # last proposal. proposal = self._proposal_roundwise[-1] self.use_non_atomic_loss = ( - isinstance(proposal.posterior_estimator._distribution, mdn) + isinstance(proposal, DirectPosterior) + and isinstance(proposal.posterior_estimator._distribution, mdn) and isinstance(self._neural_net._distribution, mdn) and check_dist_class( self._prior, class_to_check=(Uniform, MultivariateNormal)