diff --git a/ax/models/torch/alebo.py b/ax/models/torch/alebo.py index 99dad0b9a20..7d1f0a8cadf 100644 --- a/ax/models/torch/alebo.py +++ b/ax/models/torch/alebo.py @@ -398,7 +398,7 @@ def posterior( assert output_indices is None assert not observation_noise mvn = self(X) - posterior = GPyTorchPosterior(mvn=mvn) + posterior = GPyTorchPosterior(distribution=mvn) if posterior_transform is not None: return posterior_transform(posterior) return posterior