small changes

Author: bwengals

pymc3/gp/cov.py

@@ -185,7 +185,7 @@ def diag(self, X):
 
     def full(self, X, Xs=None):
         if Xs is None:
-            return tt.square(self.sigma) * tt.eye(X.shape[0])
+            return tt.diag(self.diag(X))
         else:
             return tt.alloc(0.0, X.shape[0], Xs.shape[0])
 

pymc3/gp/gp.py

@@ -502,8 +502,7 @@ def predict(self, Xnew, point=None, diag=False, pred_noise=False, given={}):
             Same as `conditional` method.
         """
         mu, cov = self.predictt(Xnew, diag, pred_noise, given)
-        mu, cov = draw_values([mu, cov], point=point)
-        return mu, cov
+        return draw_values([mu, cov], point=point)
 
     def predictt(self, Xnew, diag=False, pred_noise=False, given={}):
         R"""