Merge pull request #33 from pymc-learn/dev

Dev
pymc-learn · Dec 3, 2018 · 5d72707 · 5d72707
2 parents 2b49a97 + 8a71350
commit 5d72707
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
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,