Skip to content

Commit 7ac0432

Browse files
committed
generic get_plot_data in base.py and update prepostfit code to get hdi
1 parent bb7305a commit 7ac0432

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

causalpy/experiments/prepostfit.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -310,6 +310,7 @@ def get_plot_data_bayesian(self, hdi_prob: float = 0.94) -> pd.DataFrame:
310310
if isinstance(self.model, PyMCModel):
311311
pre_data = self.datapre.copy()
312312
post_data = self.datapost.copy()
313+
# PREDICTIONS
313314
pre_data["prediction"] = (
314315
az.extract(self.pre_pred, group="posterior_predictive", var_names="mu")
315316
.mean("sample")

0 commit comments

Comments
 (0)