diff --git a/ax/models/torch/botorch_defaults.py b/ax/models/torch/botorch_defaults.py index 81397d4cf04..05e16b203a2 100644 --- a/ax/models/torch/botorch_defaults.py +++ b/ax/models/torch/botorch_defaults.py @@ -761,7 +761,7 @@ def _get_model( gp = SingleTaskMultiFidelityGP( train_X=X, train_Y=Y, - data_fidelity=fidelity_features[0], + data_fidelities=fidelity_features[:1], input_transform=warp_tf, **kwargs, ) diff --git a/ax/storage/botorch_modular_registry.py b/ax/storage/botorch_modular_registry.py index 7dcc00c2cbc..178cb470e82 100644 --- a/ax/storage/botorch_modular_registry.py +++ b/ax/storage/botorch_modular_registry.py @@ -54,11 +54,8 @@ from botorch.models.fully_bayesian_multitask import SaasFullyBayesianMultiTaskGP # BoTorch `Model` imports -from botorch.models.gp_regression import FixedNoiseGP, SingleTaskGP -from botorch.models.gp_regression_fidelity import ( - FixedNoiseMultiFidelityGP, - SingleTaskMultiFidelityGP, -) +from botorch.models.gp_regression import SingleTaskGP +from botorch.models.gp_regression_fidelity import SingleTaskMultiFidelityGP from botorch.models.gp_regression_mixed import MixedSingleTaskGP from botorch.models.model import Model from botorch.models.model_list_gp_regression import ModelListGP @@ -110,8 +107,6 @@ # NOTE: Fixed noise models are deprecated. They point to their # supported parent classes, so that we can reap them with minimal # concern for backwards compatibility when the time comes. - FixedNoiseGP: "SingleTaskGP", - FixedNoiseMultiFidelityGP: "SingleTaskMultiFidelityGP", MixedSingleTaskGP: "MixedSingleTaskGP", ModelListGP: "ModelListGP", MultiTaskGP: "MultiTaskGP", diff --git a/tutorials/sebo.ipynb b/tutorials/sebo.ipynb index 6c527f9e3a8..059b165b504 100644 --- a/tutorials/sebo.ipynb +++ b/tutorials/sebo.ipynb @@ -61,7 +61,7 @@ "from ax.service.ax_client import AxClient, ObjectiveProperties\n", "from ax.utils.common.typeutils import checked_cast\n", "from botorch.acquisition.multi_objective import qNoisyExpectedHypervolumeImprovement\n", - "from botorch.models import FixedNoiseGP, SaasFullyBayesianSingleTaskGP, SingleTaskGP" + "from botorch.models import SaasFullyBayesianSingleTaskGP, SingleTaskGP" ] }, { @@ -247,7 +247,7 @@ "if SMOKE_TEST:\n", " N_BATCHES = 1\n", " BATCH_SIZE = 1\n", - " SURROGATE_CLASS = None # Auto-pick SingleTaskGP / FixedNoiseGP\n", + " SURROGATE_CLASS = None # Auto-pick SingleTaskGP\n", "else:\n", " N_BATCHES = 4\n", " BATCH_SIZE = 5\n",