d55bad55b6a9e97f800c97c73038bc5ed8d4b31f,arviz/data/io_pystan.py,PyStanConverter,observed_data_to_xarray,#PyStanConverter#,219

Before Change

        else:
            dims = self.dims
        if isinstance(self.observed_data, dict):
            observed_data = {}
            for key, vals in self.observed_data.items():
                vals = np.atleast_1d(vals)
                val_dims = dims.get(key)
                val_dims, coords = generate_dims_coords(
                    vals.shape, key, dims=val_dims, coords=self.coords
                )
                observed_data[key] = xr.DataArray(vals, dims=val_dims, coords=coords)
        else:
            if isinstance(self.observed_data, str):
                observed_names = [self.observed_data]
            else:

After Change

            dims = self.dims
        observed_names = self.observed_data
        if isinstance(observed_names, str):
            observed_names = [observed_names]
        observed_data = OrderedDict()
        for key in observed_names:
            vals = np.atleast_1d(posterior.data[key])
            val_dims = dims.get(key)
            val_dims, coords = generate_dims_coords(
                vals.shape, key, dims=val_dims, coords=self.coords
            )
            observed_data[key] = xr.DataArray(vals, dims=val_dims, coords=coords)
        return xr.Dataset(data_vars=observed_data, attrs=make_attrs(library=self.pystan))

    def to_inference_data(self):
        Convert all available data to an InferenceData object.

Instances