Skip to content

Commit 2144440

Browse files
SebastianAmentfacebook-github-bot
authored andcommitted
Importing SaasFullyBayesianSingleTaskGP into fit (#2705)
Summary: Pull Request resolved: #2705 Correcting land-time race condition that caused BoTorch to break on main today. Reviewed By: esantorella Differential Revision: D68727226 fbshipit-source-id: b97f52a8480acec0984a94f9313298c27eea26eb
1 parent 7a506e2 commit 2144440

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

botorch/fit.py

+4-1
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,10 @@
2222
from botorch.logging import logger
2323
from botorch.models import SingleTaskGP
2424
from botorch.models.approximate_gp import ApproximateGPyTorchModel
25-
from botorch.models.fully_bayesian import FullyBayesianSingleTaskGP
25+
from botorch.models.fully_bayesian import (
26+
FullyBayesianSingleTaskGP,
27+
SaasFullyBayesianSingleTaskGP,
28+
)
2629
from botorch.models.fully_bayesian_multitask import SaasFullyBayesianMultiTaskGP
2730
from botorch.models.map_saas import get_map_saas_model
2831
from botorch.models.model_list_gp_regression import ModelListGP

0 commit comments

Comments
 (0)