From 5a2ff909444407c7e47553548b29df08924f7e6c Mon Sep 17 00:00:00 2001
From: Kevin Stone <kevin.stone38@gmail.com>
Date: Wed, 7 Aug 2024 11:23:38 -0400
Subject: [PATCH] Updating botorch aq funcs for documentation

---
 obsidian/acquisition/acquisition_botorch.py | 19 +++++++++++--------
 1 file changed, 11 insertions(+), 8 deletions(-)

diff --git a/obsidian/acquisition/acquisition_botorch.py b/obsidian/acquisition/acquisition_botorch.py
index 245fb1c..045a027 100644
--- 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",
 ]