add classes to modules

SimonBlanke · Dec 29, 2023 · cb9dbd2 · cb9dbd2
1 parent 46d9611
commit cb9dbd2
Show file tree

Hide file tree

Showing 2 changed files with 31 additions and 0 deletions.
diff --git a/surfaces/machine_learning/machine_learning_functions/__init__.py b/surfaces/machine_learning/machine_learning_functions/__init__.py
@@ -8,3 +8,15 @@
     GradientBoostingRegressorFunction,
     KNeighborsRegressorFunction,
 )
+
+__all__ = [
+    "KNeighborsClassifierFunction",
+    "GradientBoostingRegressorFunction",
+    "KNeighborsRegressorFunction",
+]
+
+all = [
+    KNeighborsClassifierFunction,
+    GradientBoostingRegressorFunction,
+    KNeighborsRegressorFunction,
+]
diff --git a/surfaces/test_functions/__init__.py b/surfaces/test_functions/__init__.py
@@ -38,3 +38,22 @@
     "McCormickFunction",
     "BukinFunctionN6",
 ]
+
+all = [
+    SphereFunction,
+    AckleyFunction,
+    RastriginFunction,
+    RosenbrockFunction,
+    BealeFunction,
+    HimmelblausFunction,
+    HölderTableFunction,
+    CrossInTrayFunction,
+    SimionescuFunction,
+    EasomFunction,
+    BoothFunction,
+    GoldsteinPriceFunction,
+    StyblinskiTangFunction,
+    MatyasFunction,
+    McCormickFunction,
+    BukinFunctionN6,
+]