Merge pull request #1015 from EveCharbie/fix_continuity_controls

[RTR] UniquePerPhaseOPtionList -> OptionList ?

Author: pariterre

bioptim/limits/phase_transition.py

@@ -11,7 +11,7 @@
 from ..misc.enums import Node, PenaltyType, InterpolationType
 from ..misc.fcn_enum import FcnEnum
 from ..misc.mapping import BiMapping
-from ..misc.options import UniquePerPhaseOptionList
+from ..misc.options import OptionList
 
 
 from ..misc.parameters_types import (
@@ -103,7 +103,7 @@ def _get_pool_to_add_penalty(self, ocp, nlp):
             return nlp.J_internal if nlp else ocp.J_internal
 
 
-class PhaseTransitionList(UniquePerPhaseOptionList):
+class PhaseTransitionList(OptionList):
     """
     A list of PhaseTransition
 

bioptim/limits/phase_transtion_factory.py

@@ -76,11 +76,12 @@ def update_existing_transitions(self, phase_transition_list) -> list[PhaseTransi
         """Update the existing phase transitions with Mayer functions and add cyclic transitions."""
         existing_phases = []
         for pt in phase_transition_list:
-            idx_phase = pt.nodes_phase[0]
-            self.check_phase_index(idx_phase)
-            existing_phases.append(idx_phase)
-            self.update_transition_base(pt)
-            self.handle_cyclic_transition(idx_phase, pt)
+            for transition in pt:
+                idx_phase = transition.nodes_phase[0]
+                self.check_phase_index(idx_phase)
+                existing_phases.append(idx_phase)
+                self.update_transition_base(transition)
+                self.handle_cyclic_transition(idx_phase, transition)
         return self.full_phase_transitions
 
     def prepare_phase_transitions(self, phase_transition_list: PhaseTransitionList) -> list[PhaseTransition]: