diff --git a/surfaces/__init__.py b/surfaces/__init__.py index e6bcc8a..943edb0 100644 --- a/surfaces/__init__.py +++ b/surfaces/__init__.py @@ -4,3 +4,12 @@ __version__ = "0.0.7" __license__ = "MIT" + +from .mathematical_functions import mathematical_functions +from .machine_learning_functions import machine_learning_functions + + +__all__ = [ + "mathematical_functions", + "machine_learning_functions", +] diff --git a/surfaces/machine_learning_functions/__init__.py b/surfaces/machine_learning_functions/__init__.py index a7472ec..8ffe851 100644 --- a/surfaces/machine_learning_functions/__init__.py +++ b/surfaces/machine_learning_functions/__init__.py @@ -15,7 +15,7 @@ "KNeighborsRegressorFunction", ] -all = [ +machine_learning_functions = [ KNeighborsClassifierFunction, GradientBoostingRegressorFunction, KNeighborsRegressorFunction, diff --git a/surfaces/mathematical_functions/__init__.py b/surfaces/mathematical_functions/__init__.py index 5951b2e..7e2fb88 100644 --- a/surfaces/mathematical_functions/__init__.py +++ b/surfaces/mathematical_functions/__init__.py @@ -55,7 +55,7 @@ "ThreeHumpCamelFunction", ] -all = [ +mathematical_functions = [ AckleyFunction, BealeFunction, BoothFunction,