Remove REMBO & ALEBO (#2458)

Summary:
Pull Request resolved: #2458

These were marked for removal in v0.3.9. Last release was v0.4.0.

For users interested in BO with high-dimensional inputs, we recommend using SAASBO instead. See https://ax.dev/tutorials/saasbo.html for a tutorial

Reviewed By: bletham

Differential Revision: D57360740

fbshipit-source-id: d8214b935c9e22c9d690881d12897558742392b2

ax/modelbridge/registry.py

@@ -34,7 +34,6 @@
 from ax.modelbridge.random import RandomModelBridge
 from ax.modelbridge.torch import TorchModelBridge
 from ax.modelbridge.transforms.base import Transform
-from ax.modelbridge.transforms.centered_unit_x import CenteredUnitX
 from ax.modelbridge.transforms.choice_encode import (
     ChoiceToNumericChoice,
     OrderedChoiceToIntegerRange,
@@ -59,10 +58,8 @@
 from ax.models.discrete.eb_thompson import EmpiricalBayesThompsonSampler
 from ax.models.discrete.full_factorial import FullFactorialGenerator
 from ax.models.discrete.thompson import ThompsonSampler
-from ax.models.random.alebo_initializer import ALEBOInitializer
 from ax.models.random.sobol import SobolGenerator
 from ax.models.random.uniform import UniformGenerator
-from ax.models.torch.alebo import ALEBO
 from ax.models.torch.botorch import BotorchModel
 from ax.models.torch.botorch_modular.model import (
     BoTorchModel as ModularBoTorchModel,
@@ -137,9 +134,6 @@
     TaskChoiceToIntTaskChoice,
 ]
 
-ALEBO_X_trans: List[Type[Transform]] = [RemoveFixed, IntToFloat, CenteredUnitX]
-ALEBO_Y_trans: List[Type[Transform]] = [Derelativize, StandardizeY]
-
 STANDARD_TORCH_BRIDGE_KWARGS: Dict[str, Any] = {"torch_dtype": torch.double}
 
 
@@ -218,17 +212,6 @@ class ModelSetup(NamedTuple):
         transforms=ST_MTGP_trans,
         standard_bridge_kwargs=STANDARD_TORCH_BRIDGE_KWARGS,
     ),
-    "ALEBO": ModelSetup(
-        bridge_class=TorchModelBridge,
-        model_class=ALEBO,
-        transforms=ALEBO_X_trans + ALEBO_Y_trans,
-        standard_bridge_kwargs=STANDARD_TORCH_BRIDGE_KWARGS,
-    ),
-    "ALEBO_Initializer": ModelSetup(
-        bridge_class=RandomModelBridge,
-        model_class=ALEBOInitializer,
-        transforms=ALEBO_X_trans,
-    ),
     "BO_MIXED": ModelSetup(
         bridge_class=TorchModelBridge,
         model_class=ModularBoTorchModel,
@@ -444,10 +427,8 @@ class Models(ModelRegistryBase):
     MOO = "MOO"
     ST_MTGP_LEGACY = "ST_MTGP_LEGACY"
     ST_MTGP = "ST_MTGP"
-    ALEBO = "ALEBO"
     BO_MIXED = "BO_MIXED"
     ST_MTGP_NEHVI = "ST_MTGP_NEHVI"
-    ALEBO_INITIALIZER = "ALEBO_Initializer"
     CONTEXT_SACBO = "Contextual_SACBO"
 
     @classmethod