update imports

ziatdinovmax · Jan 2, 2024 · 7035dcc · 7035dcc
1 parent b0c572d
commit 7035dcc
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/gpax/acquisition/__init__.py b/gpax/acquisition/__init__.py
@@ -1,4 +1,5 @@
 from .acquisition import UCB, EI, POI, UE, Thompson, KG
 from .batch_acquisition import qEI, qPOI, qUCB, qKG
+from .optimize import optimize_acq
 
-__all__ = ["UCB", "EI", "POI", "UE", "KG", "Thompson", "qEI", "qPOI", "qUCB", "qKG"]
+__all__ = ["UCB", "EI", "POI", "UE", "KG", "Thompson", "qEI", "qPOI", "qUCB", "qKG", "optimize_acq"]