c4270b5d7d45f19278f3130ef83404fc688a7d2f,arviz/utils/xarray_utils.py,PyStanConverter,posterior_predictive_to_xarray,#PyStanConverter#,474

Before Change

    def posterior_predictive_to_xarray(self):
        Convert posterior_predictive samples to xarray.
        if isinstance(self.posterior_predictive, dict):
            data = {k : np.swapaxes(v, 0, 1)
                    for k, v in self.posterior_predictive.items()}
        else:
            dtypes = self.infer_dtypes()
            data = {}
            var_dict = self.fit.extract(self.posterior_predictive, dtypes=dtypes, permuted=False)

After Change

    def posterior_predictive_to_xarray(self):
        Convert posterior_predictive samples to xarray.
        if isinstance(self.posterior_predictive, dict):
            data = {}
            for key, values in self.posterior_predictive.items():
                if len(values.shape) == 1:
                    values = np.expand_dims(values, -1)
                values = np.swapaxes(values, 0, 1)
                data[key] = values
        else:
            dtypes = self.infer_dtypes()
            data = {}
            var_dict = self.fit.extract(self.posterior_predictive, dtypes=dtypes, permuted=False)

Instances