From 01ede162a5b4e00b3ab44f0fd1a8cdb2acdc58c6 Mon Sep 17 00:00:00 2001 From: michaeldeistler Date: Tue, 23 Apr 2024 11:48:39 +0200 Subject: [PATCH] Fixup for process_x in EnsemblePosterior --- sbi/inference/posteriors/ensemble_posterior.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/sbi/inference/posteriors/ensemble_posterior.py b/sbi/inference/posteriors/ensemble_posterior.py index 295f2085e..72af02d88 100644 --- a/sbi/inference/posteriors/ensemble_posterior.py +++ b/sbi/inference/posteriors/ensemble_posterior.py @@ -242,7 +242,9 @@ def set_default_x(self, x: Tensor) -> "NeuralPosterior": `EnsemblePosterior` that will use a default `x` when not explicitly passed. """ - self._x = x.to(self._device) + self._x = process_x( + x, x_event_shape=None, allow_iid_x=self.potential_fn.allow_iid_x + ).to(self._device) for posterior in self.posteriors: posterior.set_default_x(x)