f7e16c6997c76bd77c06ea0284a784e2ef029a5c,arviz/utils/xarray_utils.py,PyStanToNetCDF,default_varnames_coords_dims,#Any#Any#Any#,470

Before Change

            if varname not in dims:
                if len(vals.shape) == 1:
                    vals = np.expand_dims(vals, axis=1)
                vals = np.swapaxes(vals, 0, 1)
                shape_len = len(vals.shape)
                if shape_len == 2:
                    dims[varname] = []
                else:
                    dims[varname] = [
                        "{}_dim_{}".format(varname, idx)
                        for idx in range(1, shape_len-2+1)
                    ]

        return varnames, coords, dims

    def verify_coords_dims(self):
        Light checking and guessing on the structure of an xarray for a PyStan fit

After Change

            dims = {}
        for varname, dim in zip(obj.sim["pars_oi"], obj.sim["dims_oi"]):
            if varname not in dims:
                if varname == "lp__":
                    continue
                dims[varname] = [
                    "{}_dim_{}".format(varname, idx)
                    for idx in range(1, len(dim)+1)
                ]

Instances