Merge pull request #888 from pariterre/plot_server

Fixing some problem MULTIPROCESS_SERVER (#888)

Co-Authored-By: eve-mac <[email protected]>

bioptim/__init__.py

@@ -187,7 +187,7 @@
 from .limits.multinode_constraint import MultinodeConstraintFcn, MultinodeConstraintList, MultinodeConstraint
 from .limits.multinode_objective import MultinodeObjectiveFcn, MultinodeObjectiveList, MultinodeObjective
 from .limits.objective_functions import ObjectiveFcn, ObjectiveList, Objective, ParameterObjectiveList
-from .limits.path_conditions import BoundsList, InitialGuessList
+from .limits.path_conditions import BoundsList, InitialGuessList, Bounds, InitialGuess
 from .limits.fatigue_path_conditions import FatigueBounds, FatigueInitialGuess
 from .limits.penalty_controller import PenaltyController
 from .limits.penalty_helpers import PenaltyHelpers
@@ -231,4 +231,5 @@
 from .optimization.problem_type import SocpType
 from .misc.casadi_expand import lt, le, gt, ge, if_else, if_else_zero
 
+from .gui.plot import CustomPlot
 from .gui.online_callback_server import PlottingServer

bioptim/examples/getting_started/pendulum.py

@@ -150,7 +150,8 @@ def main():
 
     # --- Solve the ocp --- #
     # Default is OnlineOptim.MULTIPROCESS on Linux, OnlineOptim.MULTIPROCESS_SERVER on Windows and None on MacOS
-    sol = ocp.solve(Solver.IPOPT(show_online_optim=OnlineOptim.DEFAULT))
+    # To see the graphs on MacOS, one must run the server manually (see resources/plotting_server.py)
+    sol = ocp.solve(Solver.IPOPT(online_optim=OnlineOptim.DEFAULT))
 
     # --- Show the results graph --- #
     sol.print_cost()

bioptim/gui/online_callback_abstract.py

@@ -32,7 +32,7 @@ class OnlineCallbackAbstract(Callback, ABC):
         Send the current data to the plotter
     """
 
-    def __init__(self, ocp, opts: dict = None, show_options: dict = None):
+    def __init__(self, ocp, opts: dict = None, **show_options):
         """
         Parameters
         ----------

bioptim/gui/online_callback_multiprocess.py

@@ -24,8 +24,8 @@ class OnlineCallbackMultiprocess(OnlineCallbackAbstract):
         The multiprocessing placeholder
     """
 
-    def __init__(self, ocp, opts: dict = None, show_options: dict = None):
-        super(OnlineCallbackMultiprocess, self).__init__(ocp, opts, show_options)
+    def __init__(self, ocp, opts: dict = None, **show_options):
+        super(OnlineCallbackMultiprocess, self).__init__(ocp, opts, **show_options)
 
         self.queue = mp.Queue()
         self.plotter = self.ProcessPlotter(self.ocp)

bioptim/gui/online_callback_multiprocess_server.py

@@ -3,7 +3,7 @@
 from .online_callback_server import PlottingServer, OnlineCallbackServer
 
 
-def _start_as_multiprocess_internal(**kwargs):
+def _start_server_internal(**kwargs):
     """
     Starts the server (necessary for multiprocessing), this method should not be called directly, apart from
     run_as_multiprocess
@@ -26,7 +26,12 @@ def __init__(self, *args, **kwargs):
         """
         host = kwargs["host"] if "host" in kwargs else None
         port = kwargs["port"] if "port" in kwargs else None
-        process = Process(target=_start_as_multiprocess_internal, kwargs={"host": host, "port": port})
+        log_level = None
+        if "log_level" in kwargs:
+            log_level = kwargs["log_level"]
+            del kwargs["log_level"]
+
+        process = Process(target=_start_server_internal, kwargs={"host": host, "port": port, "log_level": log_level})
         process.start()
 
         super(OnlineCallbackMultiprocessServer, self).__init__(*args, **kwargs)