6b7ac72472c9d549dea45f35669e876c4c8ee2e9,arviz/data/io_pyro.py,PyroConverter,posterior_to_xarray,#PyroConverter#,50

Before Change

        // Do not make pyro a requirement
        from pyro.infer import EmpiricalMarginal

        try:  // Try pyro>=0.3 release syntax
            data = {
                name: utils.expand_dims(samples.enumerate_support().squeeze())
                if self.posterior.num_chains == 1
                else samples.enumerate_support().squeeze()
                for name, samples in self.posterior.marginal(
                    sites=self.latent_vars
                ).empirical.items()
            }
        except AttributeError:  // Use pyro<0.3 release syntax
            data = {}
            for var_name in self.latent_vars:
                // pylint: disable=no-member
                samples = EmpiricalMarginal(
                    self.posterior, sites=var_name
                ).get_samples_and_weights()[0]
                samples = samples.numpy().squeeze()
                data[var_name] = utils.expand_dims(samples)
        return dict_to_dataset(data, library=self.pyro, coords=self.coords, dims=self.dims)

    def observed_data_to_xarray(self):
        Convert observed data to xarray.

After Change

    def posterior_to_xarray(self):
        Convert the posterior to an xarray dataset.
        data = self.posterior.get_samples(group_by_chain=True)
        data = {k: v.detach().cpu().numpy() for k, v in data.items()}
        return dict_to_dataset(data, library=self.pyro, coords=self.coords, dims=self.dims)

    @requires("posterior")

Instances