diff --git a/aepsych/models/monotonic_rejection_gp.py b/aepsych/models/monotonic_rejection_gp.py index de61f9468..6673288bd 100644 --- a/aepsych/models/monotonic_rejection_gp.py +++ b/aepsych/models/monotonic_rejection_gp.py @@ -96,12 +96,9 @@ def __init__( dim=self.dim ) - # TODO: This allocator *must* be SobolAllocator and not the set one. This - # suggests that this model doesn't actually properly use data for inducing - # points properly. - inducing_points = SobolAllocator( - bounds=torch.stack([lb, ub]), dim=self.dim - ).allocate_inducing_points(num_inducing=self.inducing_size) + inducing_points = self.inducing_point_method.allocate_inducing_points( + num_inducing=self.inducing_size + ) inducing_points_aug = self._augment_with_deriv_index(inducing_points, 0) variational_distribution = CholeskyVariationalDistribution(