This commit is contained in:
Contramundum
2023-08-10 16:49:59 +09:00
parent 1ca3c07e9d
commit 0c35254100
@@ -199,7 +199,7 @@ class _PreferentialGP(GPyTorchModel, ExactGP):
train_x = X[mask]
weights = cnt[mask]
assert isinstance(self.likelihood, WeightedGaussianLikelihood)
assert isinstance(self.likelihood, _WeightedGaussianLikelihood)
self.likelihood.weights = weights
preferences_np = preferences.detach().numpy()