diff --git a/neps/__init__.py b/neps/__init__.py index af5a74e1..caca68e2 100644 --- a/neps/__init__.py +++ b/neps/__init__.py @@ -13,18 +13,29 @@ ) from neps.status.status import get_summary_dict, status +Integer = IntegerParameter +Float = FloatParameter +Categorical = CategoricalParameter +Constant = ConstantParameter +Architecture = ArchitectureParameter + __all__ = [ - "run", - "plot", + "Architecture", + "Integer", + "Float", + "Categorical", + "Constant", "ArchitectureParameter", "CategoricalParameter", "ConstantParameter", "FloatParameter", + "IntegerParameter", "FunctionParameter", + "run", + "plot", + "get_summary_dict", + "status", "GraphGrammar", "GraphGrammarCell", "GraphGrammarRepetitive", - "IntegerParameter", - "get_summary_dict", - "status", ]