Allow use of neps.Float instead of neps.FloatParameter

automl · Jul 3, 2024 · bef6836 · bef6836
1 parent 33f40c6
commit bef6836
Showing 1 changed file with 16 additions and 5 deletions.
diff --git 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",
 ]