Skip to content

Commit 1eab610

Browse files
committed
hdi_prob specification in get_plot_data_bayesian
1 parent 2493b17 commit 1eab610

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

causalpy/experiments/prepostfit.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -303,13 +303,21 @@ def ols_plot(self, round_to=None, **kwargs) -> tuple[plt.Figure, List[plt.Axes]]
303303

304304
return (fig, ax)
305305

306+
<<<<<<< HEAD
306307
def get_plot_data_bayesian(self, hdi_prob: float = 0.94) -> pd.DataFrame:
308+
=======
309+
def get_plot_data_bayesian(self, hdi_prob=0.94) -> pd.DataFrame:
310+
>>>>>>> 1734486 (hdi_prob specification in get_plot_data_bayesian)
307311
"""
308312
Recover the data of a PrePostFit experiment along with the prediction and causal impact information.
309313
"""
310314
if isinstance(self.model, PyMCModel):
311315
pre_data = self.datapre.copy()
312316
post_data = self.datapost.copy()
317+
<<<<<<< HEAD
318+
=======
319+
320+
>>>>>>> 1734486 (hdi_prob specification in get_plot_data_bayesian)
313321
pre_data["prediction"] = (
314322
az.extract(self.pre_pred, group="posterior_predictive", var_names="mu")
315323
.mean("sample")

0 commit comments

Comments
 (0)