Skip to content

Commit

Permalink
remove extra copy of log-likelihood (#2347)
Browse files Browse the repository at this point in the history
Co-authored-by: andrewdipper <[email protected]>
  • Loading branch information
andrewdipper and andrewdipper committed May 10, 2024
1 parent ab82f80 commit 0f71f9e
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion arviz/data/io_numpyro.py
Original file line number Diff line number Diff line change
Expand Up @@ -194,7 +194,7 @@ def log_likelihood_to_xarray(self):
)
for obs_name, log_like in log_likelihood_dict.items():
shape = (self.nchains, self.ndraws) + log_like.shape[1:]
data[obs_name] = np.reshape(log_like.copy(), shape)
data[obs_name] = np.reshape(np.asarray(log_like), shape)
return dict_to_dataset(
data,
library=self.numpyro,
Expand Down

0 comments on commit 0f71f9e

Please sign in to comment.