overhaul of __init__.py

Author: alphaville

python/opengen/builder/__init__.py

@@ -1,3 +1,12 @@
-from .optimizer_builder import *
-from .problem import *
-from .set_y_calculator import *
+from .optimizer_builder import OpEnOptimizerBuilder
+from .problem import Problem
+from .set_y_calculator import SetYCalculator
+from .ros_builder import RosBuilder, ROS2Builder
+
+__all__ = [
+	"OpEnOptimizerBuilder",
+	"Problem",
+	"SetYCalculator",
+	"RosBuilder",
+	"ROS2Builder",
+]

python/opengen/builder/optimizer_builder.py

@@ -2,6 +2,7 @@
 import shutil
 import yaml
 
+import opengen as og
 import opengen.config as og_cfg
 import opengen.definitions as og_dfn
 import opengen.constraints as og_cstr
@@ -43,10 +44,10 @@ class OpEnOptimizerBuilder:
     """
 
     def __init__(self,
-                 problem,
-                 metadata=og_cfg.OptimizerMeta(),
-                 build_configuration=og_cfg.BuildConfiguration(),
-                 solver_configuration=og_cfg.SolverConfiguration()):
+                 problem: og.builder.Problem,
+                 metadata: og_cfg.OptimizerMeta =og_cfg.OptimizerMeta(),
+                 build_configuration: og_cfg.BuildConfiguration =og_cfg.BuildConfiguration(),
+                 solver_configuration: og_cfg.SolverConfiguration=og_cfg.SolverConfiguration()):
         """Constructor of OpEnOptimizerBuilder
 
         :param problem: instance of :class:`~opengen.builder.problem.Problem`
@@ -645,7 +646,7 @@ def __initialize(self):
 
     def __check_user_provided_parameters(self):
         self.__logger.info("Checking user parameters")
-        
+
         # Check constraints dimensions
         dim_constraints = self.__problem.constraints.dimension()        
         dim_decision_variables = self.__problem.dim_decision_variables()

python/opengen/config/__init__.py

@@ -1,5 +1,15 @@
-from .meta import *
-from .solver_config import *
-from .build_config import *
-from .tcp_server_config import *
-from .ros_config import *
+from .meta import OptimizerMeta, SEMVER_PATTERN
+from .solver_config import SolverConfiguration
+from .build_config import BuildConfiguration, RustAllocator
+from .tcp_server_config import TcpServerConfiguration
+from .ros_config import RosConfiguration
+
+__all__ = [
+	"OptimizerMeta",
+	"SEMVER_PATTERN",
+	"SolverConfiguration",
+	"BuildConfiguration",
+	"RustAllocator",
+	"TcpServerConfiguration",
+	"RosConfiguration",
+]

python/opengen/constraints/__init__.py

@@ -1,14 +1,31 @@
-from .ball1 import *
-from .ball2 import *
-from .sphere2 import *
-from .rectangle import *
-from .constraint import *
-from .ball_inf import *
-from .soc import *
-from .no_constraints import *
-from .cartesian import *
-from .zero import *
-from .finite_set import *
-from .halfspace import *
-from .simplex import *
-from .affine_space import *
+from .ball1 import Ball1
+from .ball2 import Ball2
+from .sphere2 import Sphere2
+from .rectangle import Rectangle
+from .constraint import Constraint
+from .ball_inf import BallInf
+from .soc import SecondOrderCone
+from .no_constraints import NoConstraints
+from .cartesian import CartesianProduct
+from .zero import Zero
+from .finite_set import FiniteSet
+from .halfspace import Halfspace
+from .simplex import Simplex
+from .affine_space import AffineSpace
+
+__all__ = [
+	"Ball1",
+	"Ball2",
+	"Sphere2",
+	"Rectangle",
+	"Constraint",
+	"BallInf",
+	"SecondOrderCone",
+	"NoConstraints",
+	"CartesianProduct",
+	"Zero",
+	"FiniteSet",
+	"Halfspace",
+	"Simplex",
+	"AffineSpace",
+]

python/opengen/functions/__init__.py

@@ -1,9 +1,21 @@
-from .rosenbrock import *
-from .fmin import *
-from .fmax import *
-from .is_symbolic import *
-from .is_numeric import *
-from .sign import *
-from .norm2 import *
-from .fabs import *
-from .norm2_squared import *
+from .rosenbrock import rosenbrock
+from .fmin import fmin
+from .fmax import fmax
+from .is_symbolic import is_symbolic
+from .is_numeric import is_numeric
+from .sign import sign
+from .norm2 import norm2
+from .fabs import fabs
+from .norm2_squared import norm2_squared
+
+__all__ = [
+	"rosenbrock",
+	"fmin",
+	"fmax",
+	"is_symbolic",
+	"is_numeric",
+	"sign",
+	"norm2",
+	"fabs",
+	"norm2_squared",
+]

python/opengen/tcp/__init__.py

@@ -1,4 +1,11 @@
-from .optimizer_tcp_manager import *
-from .solver_status import *
-from .solver_error import *
-from .solver_response import *
+from .optimizer_tcp_manager import OptimizerTcpManager
+from .solver_status import SolverStatus
+from .solver_error import SolverError
+from .solver_response import SolverResponse
+
+__all__ = [
+	"OptimizerTcpManager",
+	"SolverStatus",
+	"SolverError",
+	"SolverResponse",
+]

python/test/__init__.py

@@ -0,0 +1,3 @@
+"""Test package for the project."""
+
+__all__ = []