From 1237c92987c4f8018cb1af15d513b5bea29830b1 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Jo=C3=A3o=20Faria?= <joao.faria@astro.up.pt>
Date: Wed, 9 Sep 2020 15:28:27 +0100
Subject: [PATCH] reverting some changes in GP prediction

---
 pykima/GP.py | 8 ++------
 1 file changed, 2 insertions(+), 6 deletions(-)

diff --git a/pykima/GP.py b/pykima/GP.py
index 7887d5f1..a7370905 100644
--- 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):
         """