Add initial guess upon defining optimization.

Author: javiarrobas

wedoco_optimo/model.py

@@ -144,6 +144,7 @@ def initialize_optimization(self, ipopt_options: dict=None, prescribed_inputs: d
 
     def define_optimization(self,
                             x0: np.array=None,
+                            initial_guess: dict=None,
                             constraints: dict=None,
                             objective_terms: list=None, 
                             ipopt_options: dict=None,
@@ -164,6 +165,11 @@ def define_optimization(self,
         for x_name in self.dae.x():
             self.ocp.set_initial(self.dae.var(x_name), x0[x_name])
 
+        # Set the initial guess for other provided variables
+        if initial_guess is not None:
+            for v_name in initial_guess:
+                self.ocp.set_initial(self.dae.var(v_name), initial_guess[v_name])
+
         # Set initial time
         t0 = 0
         self.ocp.set_t0(0)