Skip to content

Commit 079e131

Browse files
committed
doc: update docstrings
1 parent a0501fe commit 079e131

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

pymc_extras/model_builder.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -543,6 +543,9 @@ def predict(
543543
The input data used for prediction.
544544
extend_idata : Boolean determining whether the predictions should be added to inference data object.
545545
Defaults to True.
546+
predictions : bool
547+
Whether to use the predictions group for posterior predictive sampling.
548+
Defaults to True.
546549
**kwargs: Additional arguments to pass to pymc.sample_posterior_predictive
547550
548551
Returns
@@ -651,7 +654,6 @@ def sample_posterior_predictive(self, X_pred, extend_idata, predictions, combine
651654
if extend_idata:
652655
self.idata.extend(post_pred, join="right")
653656

654-
# Determine the correct group
655657
group_name = "predictions" if predictions else "posterior_predictive"
656658

657659
posterior_predictive_samples = az.extract(
@@ -718,6 +720,8 @@ def predict_posterior(
718720
Defaults to True.
719721
combined: Combine chain and draw dims into sample. Won't work if a dim named sample already exists.
720722
Defaults to True.
723+
predictions : Boolean determing whether to use the predictions group for posterior predictive sampling.
724+
Defaults to True.
721725
**kwargs: Additional arguments to pass to pymc.sample_posterior_predictive
722726
723727
Returns

0 commit comments

Comments
 (0)