Fixup for process_x in EnsemblePosterior (#1148)

sbi-dev · Apr 23, 2024 · 005aeac · 005aeac
1 parent f0aa0f9
commit 005aeac
Showing 1 changed file with 3 additions and 1 deletion.
diff --git 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)