feat: improve imports

automl · Jan 15, 2025 · da1b7d8 · da1b7d8
1 parent 40ad1f1
commit da1b7d8
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/neps/__init__.py b/neps/__init__.py
@@ -1,16 +1,25 @@
 from neps.api import run
+from neps.optimizers import algorithms
+from neps.optimizers.ask_and_tell import AskAndTell
+from neps.optimizers.optimizer import SampledConfig
 from neps.plot.plot import plot
 from neps.plot.tensorboard_eval import tblogger
 from neps.space import Categorical, Constant, Float, Integer, SearchSpace
+from neps.state import BudgetInfo, Trial
 from neps.status.status import get_summary_dict, status
 from neps.utils.files import load_and_merge_yamls as load_yamls
 
 __all__ = [
+    "AskAndTell",
+    "BudgetInfo",
     "Categorical",
     "Constant",
     "Float",
     "Integer",
+    "SampledConfig",
     "SearchSpace",
+    "Trial",
+    "algorithms",
     "get_summary_dict",
     "load_yamls",
     "plot",