Update sbi/inference/posteriors/ensemble_posterior.py

Co-authored-by: Jan <[email protected]>

Author: CompiledAtBirth

sbi/inference/posteriors/ensemble_posterior.py

@@ -137,7 +137,7 @@ def weights(self, weights: Optional[Union[List[float], Tensor]]) -> None:
             self._weights = torch.tensor([
                 1.0 / self.num_components for _ in range(self.num_components)
             ])
-        elif isinstance(weights, Tensor) or isinstance(weights, List):
+        elif isinstance(weights, (Tensor, List)):
             self._weights = torch.tensor(weights) / sum(weights)
         else:
             raise TypeError