reverting some changes in GP prediction

j-faria · Sep 9, 2020 · 1237c92 · 1237c92
1 parent 80cbdfe
commit 1237c92
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/pykima/GP.py b/pykima/GP.py
@@ -153,10 +153,7 @@ def predict_with_hyperpars(self, results, sample, t=None, add_parts=True,
             pred = self.predict(y, t, return_cov=False, return_std=return_std)
             return pred
 
-        mu = self.predict(y, results.t, return_cov=False,
-                          return_std=return_std)
-        if return_std:
-            mu, std = mu
+        mu = self.predict(y, results.t, return_cov=False)
 
         if add_parts:
             # add the trend back
@@ -167,8 +164,7 @@ def predict_with_hyperpars(self, results, sample, t=None, add_parts=True,
                 for i in range(1, results.n_instruments):
                     mu[results.obs == i] += sample[results.indices['inst_offsets']][i-1]
 
-        return mu, std
-
+        return mu
 
     def sample_conditional(self, y, t, size=1):
         """