From e67a09b4d8eba2344f3ca81de481139554e5b0ff Mon Sep 17 00:00:00 2001 From: Sebastian Ament Date: Mon, 27 Jan 2025 13:36:13 -0800 Subject: [PATCH] Importing `SaasFullyBayesianSingleTaskGP` into `fit` (#2705) Summary: Correcting land-time race condition that caused BoTorch to break on main today. Reviewed By: esantorella Differential Revision: D68727226 --- botorch/fit.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/botorch/fit.py b/botorch/fit.py index 0c045beeba..ad250a5239 100644 --- a/botorch/fit.py +++ b/botorch/fit.py @@ -22,7 +22,10 @@ from botorch.logging import logger from botorch.models import SingleTaskGP from botorch.models.approximate_gp import ApproximateGPyTorchModel -from botorch.models.fully_bayesian import FullyBayesianSingleTaskGP +from botorch.models.fully_bayesian import ( + FullyBayesianSingleTaskGP, + SaasFullyBayesianSingleTaskGP, +) from botorch.models.fully_bayesian_multitask import SaasFullyBayesianMultiTaskGP from botorch.models.map_saas import get_map_saas_model from botorch.models.model_list_gp_regression import ModelListGP