add constant mean function for gpr

Emaasit · Emaasit · commit abde49e7e84f · 2018-12-03T11:20:29.000-05:00
diff --git a/pmlearn/__init__.py b/pmlearn/__init__.py
@@ -10,7 +10,7 @@
 See http://pymc-learn.org for complete documentation.
 """
 
-__version__ = '0.0.1.rc2'
+__version__ = '0.0.1.rc3'
 
 __all__ = ['gaussian_process',
            'linear_model',
diff --git a/pmlearn/gaussian_process/gpr.py b/pmlearn/gaussian_process/gpr.py
@@ -144,7 +144,7 @@ def create_model(self):
             if self.prior_mean is None:
                 mean_function = pm.gp.mean.Zero()
             else:
-                mean_function = self.prior_mean
+                mean_function = pm.gp.mean.Constant(c=self.prior_mean)
 
             self.gp = pm.gp.Latent(mean_func=mean_function,
                                    cov_func=cov_function)
@@ -259,7 +259,7 @@ def create_model(self):
             if self.prior_mean is None:
                 mean_function = pm.gp.mean.Zero()
             else:
-                mean_function = self.prior_mean
+                mean_function = pm.gp.mean.Constant(c=self.prior_mean)
 
             self.gp = pm.gp.Latent(mean_func=mean_function,
                                    cov_func=cov_function)
@@ -373,7 +373,7 @@ def create_model(self):
             if self.prior_mean is None:
                 mean_function = pm.gp.mean.Zero()
             else:
-                mean_function = self.prior_mean
+                mean_function = pm.gp.mean.Constant(c=self.prior_mean)
 
             self.gp = pm.gp.MarginalSparse(mean_func=mean_function,
                                            cov_func=cov_function,
