..

Author: dweindl

petab/v2/converters.py

@@ -41,6 +41,12 @@ class ExperimentsToEventsConverter:
     #: ID of the parameter that indicates whether the model is in
     #  the pre-equilibration phase (1) or not (0).
     PREEQ_INDICATOR = "_petab_preequilibration_indicator"
+    #: The condition ID of the condition that sets the
+    #: pre-equilibration indicator to 1.
+    CONDITION_ID_PREEQ_ON = "_petab_preequilibration_on"
+    #: The condition ID of the condition that sets the
+    #: pre-equilibration indicator to 0.
+    CONDITION_ID_PREEQ_OFF = "_petab_preequilibration_off"
 
     def __init__(self, problem: Problem):
         """Initialize the converter.
@@ -220,17 +226,17 @@ def _convert_experiment(self, experiment: Experiment):
         for period in kept_periods:
             period.condition_ids = [
                 self._get_experiment_indicator_condition_id(experiment.id),
-                "_petab_preequilibration"
+                self.CONDITION_ID_PREEQ_ON
                 if period.time == TIME_PREEQUILIBRATION
-                else "_petab_no_preequilibration",
+                else self.CONDITION_ID_PREEQ_OFF,
             ]
 
         experiment.periods = kept_periods
 
     def _create_period_start_event(
         self, experiment: Experiment, i_period: int, period: ExperimentPeriod
     ) -> libsbml.Event:
-        """Create an event that triggers at the beginning of a period."""
+        """Create an event that triggers at the start of a period."""
 
         # TODO: for now, add separate events for each experiment x period,
         #  this could be optimized to reuse events
@@ -340,15 +346,15 @@ def _add_indicators_to_conditions(self) -> None:
         # create conditions for indicator parameters
         problem.condition_table.conditions.append(
             Condition(
-                id="_petab_preequilibration",
+                id=self.CONDITION_ID_PREEQ_ON,
                 changes=[
                     Change(target_id=self._preeq_indicator, target_value=1)
                 ],
             )
         )
         problem.condition_table.conditions.append(
             Condition(
-                id="_petab_no_preequilibration",
+                id=self.CONDITION_ID_PREEQ_OFF,
                 changes=[
                     Change(target_id=self._preeq_indicator, target_value=0)
                 ],