Updating botorch aq funcs for documentation

MSDLLCpapers · Aug 7, 2024 · 5a2ff90 · 5a2ff90
1 parent 22bae69
commit 5a2ff90
Showing 1 changed file with 11 additions and 8 deletions.
diff --git a/obsidian/acquisition/acquisition_botorch.py b/obsidian/acquisition/acquisition_botorch.py
@@ -11,22 +11,25 @@
 from botorch.acquisition.objective import MCAcquisitionObjective, PosteriorTransform
 
 # Botorch acquisition functions supported in obsidian
-from botorch.acquisition import qExpectedImprovement, qProbabilityOfImprovement, \
-    qUpperConfidenceBound, qNoisyExpectedImprovement, qSimpleRegret
-from botorch.acquisition.multi_objective.monte_carlo import qNoisyExpectedHypervolumeImprovement
+from botorch.acquisition import qProbabilityOfImprovement, qUpperConfidenceBound, qSimpleRegret
+from botorch.acquisition.logei import qLogExpectedImprovement, qLogNoisyExpectedImprovement
+from botorch.acquisition.multi_objective.logei import qLogExpectedHypervolumeImprovement, qLogNoisyExpectedHypervolumeImprovement
 from botorch.acquisition.multi_objective.parego import qLogNParEGO
+from botorch.acquisition.active_learning import qNegIntegratedPosteriorVariance
 
 
 __all__ = [
-    "qExpectedImprovement",
+    "qMean",
+    "qSpaceFill",
     "qProbabilityOfImprovement",
     "qUpperConfidenceBound",
-    "qNoisyExpectedImprovement",
     "qSimpleRegret",
-    "qNoisyExpectedHypervolumeImprovement",
+    "qLogExpectedImprovement",
+    "qLogNoisyExpectedImprovement",
+    "qLogExpectedHypervolumeImprovement",
+    "qLogNoisyExpectedHypervolumeImprovement",
     "qLogNParEGO",
-    "qMean",
-    "qSpaceFill",
+    "qNegIntegratedPosteriorVariance",
 ]