From 89ae4b403c083de3109be88ad057932b531e0927 Mon Sep 17 00:00:00 2001 From: Elizabeth Santorella Date: Tue, 19 Mar 2024 14:20:24 -0700 Subject: [PATCH] Replace calls to deprecated `mvn` argument of `GPyTorchPosterior` with `distribution` (#2289) Summary: This will not change behavior. Reviewed By: saitcakmak Differential Revision: D55094724 --- ax/models/torch/alebo.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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