Use human-readable indexes for xarray object

bambinos · Jun 13, 2022 · 209992d · 209992d
1 parent 4e288c2
commit 209992d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/bambi/models.py b/bambi/models.py
@@ -953,7 +953,7 @@ def predict_sklearn(self, idata, predict_variable, data=None):
         ndarray
         """
         self.predict(idata, kind="pps", data=data)
-        return idata.posterior_predictive[predict_variable].mean(axis=(0, 1)).values
+        return idata.posterior_predictive[predict_variable].mean(("chain", "draw")).values
 
     def graph(self, formatting="plain", name=None, figsize=None, dpi=300, fmt="png"):
         """