Skip to content

Commit

Permalink
Remove FixedNoiseMultiTaskGP import
Browse files Browse the repository at this point in the history
Summary:
Remove the `FixedNoiseMultiTaskGP` import to unblock its full deprecation.

Note: The storage will still support loading these models by converting them to `MultiTaskGP`.

Reviewed By: Balandat

Differential Revision: D56802232
  • Loading branch information
saitcakmak authored and facebook-github-bot committed May 1, 2024
1 parent 5ae5ceb commit e6c65cd
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions ax/storage/botorch_modular_registry.py
Original file line number Diff line number Diff line change
Expand Up @@ -61,7 +61,7 @@
from botorch.models.gp_regression_mixed import MixedSingleTaskGP
from botorch.models.model import Model
from botorch.models.model_list_gp_regression import ModelListGP
from botorch.models.multitask import FixedNoiseMultiTaskGP, MultiTaskGP
from botorch.models.multitask import MultiTaskGP
from botorch.models.transforms.input import (
ChainedInputTransform,
InputPerturbation,
Expand Down Expand Up @@ -108,7 +108,6 @@
# concern for backwards compatibility when the time comes.
FixedNoiseGP: "SingleTaskGP",
FixedNoiseMultiFidelityGP: "SingleTaskMultiFidelityGP",
FixedNoiseMultiTaskGP: "MultiTaskGP",
MixedSingleTaskGP: "MixedSingleTaskGP",
ModelListGP: "ModelListGP",
MultiTaskGP: "MultiTaskGP",
Expand Down Expand Up @@ -208,6 +207,7 @@


REVERSE_MODEL_REGISTRY: Dict[str, Type[Model]] = {
# NOTE: These ensure backwards compatibility. Keep them around.
"FixedNoiseGP": SingleTaskGP,
"FixedNoiseMultiFidelityGP": SingleTaskMultiFidelityGP,
"FixedNoiseMultiTaskGP": MultiTaskGP,
Expand Down

0 comments on commit e6c65cd

Please sign in to comment.