code: def sample_stats_to_xarray(self):
          
          posterior = self.posterior
          posterior_model = self.posterior_model
          # copy dims and coords
          dims = deepcopy(self.dims) if self.dims is not None else {}
          coords = deepcopy(self.coords) if self.coords is not None else {}
  
          # log_likelihood
          log_likelihood = self.log_likelihood
          if log_likelihood is not None:
              if isinstance(log_likelihood, str) and log_likelihood in dims:
                  dims["log_likelihood"] = dims.pop(log_likelihood)
  
          data = get_sample_stats_stan3(
              posterior, model=posterior_model, log_likelihood=log_likelihood
          )
  
          return dict_to_dataset(data, library=self.stan, coords=coords, dims=dims)
docs: Extract sample_stats from posterior.