From af543fe2872a284a48500fc8c38e3cad33566c83 Mon Sep 17 00:00:00 2001 From: Jelena Markovic-Voronov Date: Wed, 27 Sep 2023 16:42:03 -0700 Subject: [PATCH] FullyBayesian SAAS model migration (#1879) Summary: Pull Request resolved: https://github.com/facebook/Ax/pull/1879 FullyBayesian high-dim model changed from legacy to MBM for single and multi-objective without input transforms. Reviewed By: saitcakmak Differential Revision: D49479004 fbshipit-source-id: a9df6312be597c7e77f879dc2843cbbcf03bce85 --- ax/models/torch/botorch_modular/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ax/models/torch/botorch_modular/utils.py b/ax/models/torch/botorch_modular/utils.py index c793d1aa4e4..2223a1467f1 100644 --- a/ax/models/torch/botorch_modular/utils.py +++ b/ax/models/torch/botorch_modular/utils.py @@ -303,7 +303,7 @@ def fit_botorch_model( fit_fully_bayesian_model_nuts( m, disable_progbar=True, - jit_compile=mll_options.get("jit_compile", False), + **mll_options, ) elif isinstance(m, (GPyTorchModel, PairwiseGP)): mll_options = mll_options or {}