From 307d9e01772bc377b90935f756dee584b6f07363 Mon Sep 17 00:00:00 2001 From: Arno Solin Date: Wed, 2 Nov 2022 17:30:01 +0200 Subject: [PATCH] Resolve issue that broke tests for returning var in spacetime --- bayesnewton/basemodels.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/bayesnewton/basemodels.py b/bayesnewton/basemodels.py index 4104653..cff2fea 100644 --- a/bayesnewton/basemodels.py +++ b/bayesnewton/basemodels.py @@ -813,7 +813,7 @@ def predict(self, X=None, R=None, pseudo_lik_params=None): # if np.squeeze(test_var).ndim > 2: # deal with spatio-temporal case (discard spatial covariance) if self.spatio_temporal: # deal with spatio-temporal case (discard spatial covariance) - test_var = diag(np.squeeze(test_var)) + test_var = diag(test_var) return np.squeeze(test_mean), np.squeeze(test_var) def filter_energy(self):