remove empty dimension from Constant

Author: bwengals

pymc3/gp/mean.py

@@ -64,7 +64,7 @@ def __init__(self, coeffs, intercept=0):
         self.A = coeffs
 
     def __call__(self, X):
-        return tt.dot(X, self.A) + self.b
+        return (tt.dot(X, self.A) + self.b).squeeze()
 
 
 class Add(Mean):

pymc3/tests/test_gp.py

@@ -34,7 +34,8 @@ def test_value(self):
         with Model() as model:
             linear_mean = gp.mean.Linear(2, 0.5)
         M = theano.function([], linear_mean(X))()
-        npt.assert_allclose(M[1, 0], 0.7222, atol=1e-3)
+        npt.assert_allclose(M[1], 0.7222, atol=1e-3)
+        assert M.shape == (10, )
 
 
 class TestAddProdMean(object):