86b670cf47e4442af93bfe1621ca4a9ef1220757,arviz/utils/xarray_utils.py,PyStanToXarray,to_xarray,#PyStanToXarray#,393

Before Change

        for key in self.varnames:
            var_dtype = {key : "int"} if key in dtypes else {}
            vals = fit.extract(key, dtypes=var_dtype, permuted=False)[key]
            if len(vals.shape) == 1:
                vals = np.expand_dims(vals, axis=1)
            vals = np.swapaxes(vals, 0, 1)
            dims_str = base_dims + self.dims[key]

After Change

        data = xr.Dataset(coords=self.coords)
        base_dims = ["chain", "draw"]
        
        extract = fit.extract(pars=self.varnames,
                              dtypes=dtypes,
                              permuted=False)
        for key, vals in extract.items():
            if len(vals.shape) == 1:
                vals = np.expand_dims(vals, axis=1)
            vals = np.swapaxes(vals, 0, 1)

Instances