From 0f71f9e12353a399cdd25138913e2fb73251fbcb Mon Sep 17 00:00:00 2001 From: andrewdipper Date: Fri, 10 May 2024 02:11:35 -0700 Subject: [PATCH] remove extra copy of log-likelihood (#2347) Co-authored-by: andrewdipper --- arviz/data/io_numpyro.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/arviz/data/io_numpyro.py b/arviz/data/io_numpyro.py index e742732ee2..f8fc098a68 100644 --- a/arviz/data/io_numpyro.py +++ b/arviz/data/io_numpyro.py @@ -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,