Add None for fields not used

arviz-devs · Nov 3, 2022 · d2604da · d2604da
1 parent 8c819d2
commit d2604da
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/arviz/data/io_beanmachine.py b/arviz/data/io_beanmachine.py
@@ -24,15 +24,23 @@ def __init__(
 
         if "posterior" in self.sampler.namespaces:
             self.posterior = self.sampler.namespaces["posterior"].samples
+        else:
+            self.posterior = None
 
         if "posterior_predictive" in self.sampler.namespaces:
             self.posterior_predictive = self.sampler.namespaces["posterior_predictive"].samples
+        else:
+            self.posterior_predictive = None
 
         if self.sampler.log_likelihoods is not None:
             self.log_likelihoods = self.sampler.log_likelihoods
+        else:
+            self.log_likelihoods = None
 
         if self.sampler.observations is not None:
             self.observations = self.sampler.observations
+        else:
+            self.observations = None
 
     @requires("posterior")
     def posterior_to_xarray(self):