From 99d3e36e9ffae999383604f6078f218bb64640da Mon Sep 17 00:00:00 2001 From: Sait Cakmak Date: Fri, 23 Aug 2024 13:50:10 -0700 Subject: [PATCH] Update import paths for base multi-objective acquisition functions Summary: As titled. Follow up to D61721280. Reviewed By: esantorella Differential Revision: D61733064 --- ax/models/torch/utils.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/ax/models/torch/utils.py b/ax/models/torch/utils.py index 9f986026e3a..4c8cc894df0 100644 --- a/ax/models/torch/utils.py +++ b/ax/models/torch/utils.py @@ -26,10 +26,8 @@ qSimpleRegret, SampleReducingMCAcquisitionFunction, ) -from botorch.acquisition.multi_objective.analytic import ( +from botorch.acquisition.multi_objective.base import ( MultiObjectiveAnalyticAcquisitionFunction, -) -from botorch.acquisition.multi_objective.monte_carlo import ( MultiObjectiveMCAcquisitionFunction, ) from botorch.acquisition.multi_objective.multi_output_risk_measures import (