diff --git a/botorch/models/__init__.py b/botorch/models/__init__.py index 06bd17b5b1..a5e6dd44b8 100644 --- a/botorch/models/__init__.py +++ b/botorch/models/__init__.py @@ -27,6 +27,9 @@ from botorch.models.model_list_gp_regression import ModelListGP from botorch.models.multitask import KroneckerMultiTaskGP, MultiTaskGP from botorch.models.pairwise_gp import PairwiseGP, PairwiseLaplaceMarginalLogLikelihood +from botorch.models.robust_relevance_pursuit_model import ( + RobustRelevancePursuitSingleTaskGP, +) __all__ = [ "add_saas_prior", @@ -49,4 +52,5 @@ "SingleTaskGP", "SingleTaskMultiFidelityGP", "SingleTaskVariationalGP", + "RobustRelevancePursuitSingleTaskGP", ]