self.gp_observation_model = gp_observation_model
// Buffers for conditioning on dataif train_xs is not None and train_y is not None:
self.update_data(train_xs, train_y)
def update_data(self, train_xs, train_y):
ifisinstance(train_xs, Variable) or isinstance(train_xs, torch._TensorBase):
train_xs = (train_xs,)
train_xs = [input.data ifisinstance(input, Variable)else input for input in train_xs]