Fix gpmc _compile() to support optimiser and return `opt_step`
@@ -52,7 +52,7 @@ class GPMC(GPModel):
self.V = Param(np.zeros((self.num_data, self.num_latent)))
self.V.prior = Gaussian(0., 1.)
- def _compile(self):
+ def _compile(self, optimizer=None):
"""
Before calling the standard compile function, check to see if the size
of the data has changed and add parameters appropriately.
@@ -64,7 +64,8 @@ class GPMC(GPModel):
self.num_data = self.X.shape[0]
self.V = Param(np.zeros((self.num_data, self.num_latent)))
self.V.prior = Gaussian(0., 1.)
- super(GPMC, self)._compile()
+
+ return super(GPMC, self)._compile(optimizer=optimizer)
def build_likelihood(self):
"""
