diff --git a/ax/models/torch/botorch_defaults.py b/ax/models/torch/botorch_defaults.py index 862510a476e..1e476a7e6d5 100644 --- a/ax/models/torch/botorch_defaults.py +++ b/ax/models/torch/botorch_defaults.py @@ -772,6 +772,7 @@ def _get_model( train_Yvar=None if all_nan_Yvar else Yvar, covar_module=covar_module, input_transform=warp_tf, + outcome_transform=None, # For backwards compatibility **kwargs, ) else: diff --git a/ax/models/torch/botorch_modular/surrogate.py b/ax/models/torch/botorch_modular/surrogate.py index 568e99b2c11..240873b128c 100644 --- a/ax/models/torch/botorch_modular/surrogate.py +++ b/ax/models/torch/botorch_modular/surrogate.py @@ -358,6 +358,17 @@ def _set_formatted_inputs( ) -> None: for input_name, input_class, input_options in inputs: if input_class is None: + # This is a temporary solution until all BoTorch models use + # `Standardize` by default, see TODO [T197435440]. + # After this, we should update `Surrogate` to use `DEFAULT` + # (https://fburl.com/code/22f4397e) for both of these args. This will + # allow users to explicitly disable the default transforms by passing + # in `None`. + if ( + input_name in ["outcome_transform"] + and input_name in botorch_model_class_args + ): + formatted_model_inputs[input_name] = None continue if input_name not in botorch_model_class_args: # TODO: We currently only pass in `covar_module` and `likelihood`