Skip to content

Commit c1baf2f

Browse files
committed
rename _posterior_to_idata and add docstring
1 parent c36cdbc commit c1baf2f

File tree

1 file changed

+14
-3
lines changed

1 file changed

+14
-3
lines changed

bambi/backend/pymc.py

+14-3
Original file line numberDiff line numberDiff line change
@@ -473,11 +473,22 @@ def _run_laplace(self, draws):
473473

474474
samples = np.random.multivariate_normal(modes, cov, size=draws)
475475

476-
return _posterior_to_idata(samples, self.model)
476+
return _posterior_samples_to_idata(samples, self.model)
477477

478478

479-
def _posterior_to_idata(samples, model):
480-
""" """
479+
def _posterior_samples_to_idata(samples, model):
480+
"""Create InferenceData from samples.
481+
482+
Parameters
483+
----------
484+
samples: array
485+
Posterior samples
486+
model: PyMC model
487+
488+
Returns
489+
-------
490+
An ArviZ's InferenceData object.
491+
"""
481492
initial_point = model.initial_point(seed=None)
482493
variables = model.value_vars
483494

0 commit comments

Comments
 (0)