Updates the training of LineBO to botorch==0.11.0 (#31)

MachineLearningLifeScience · May 26, 2024 · 0d1b030 · 0d1b030
1 parent 95e77a5
commit 0d1b030
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/...es/solvers/bayesian_optimization/base_bayesian_optimization/base_bayesian_optimization.py b/...es/solvers/bayesian_optimization/base_bayesian_optimization/base_bayesian_optimization.py
@@ -10,7 +10,6 @@
 from botorch.fit import (
     fit_gpytorch_mll_torch,
     fit_gpytorch_mll_scipy,
-    fit_gpytorch_model,
 )
 from botorch.acquisition import (
     ExpectedImprovement,
@@ -171,7 +170,7 @@ def _fit_model(
             covar_module=self.kernel,
         )
         mll = ExactMarginalLogLikelihood(model_instance.likelihood, model_instance)
-        fit_gpytorch_model(mll)
+        fit_gpytorch_mll_torch(mll)
         model_instance.eval()
 
         self.gp_model_of_objective = model_instance