Importing SaasFullyBayesianSingleTaskGP into fit

Summary: Correcting land-time race condition that caused BoTorch to break on main today. Differential Revision: D68727226
pytorch · Jan 27, 2025 · cf83469 · cf83469
1 parent 90fc872
commit cf83469
Showing 1 changed file with 4 additions and 1 deletion.
diff --git 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