Fix missing fields

Author: akalpokas

src/somd2/config/_config.py

@@ -1024,6 +1024,7 @@ def lambda_schedule(self, lambda_schedule):
                     self._lambda_schedule = _LambdaSchedule.charge_scaled_morph(0.2)
                     self._lambda_schedule_name = "charge_scaled_morph"
                 elif lambda_schedule == "ring_break_morph":
+                    self._lambda_schedule = _LambdaSchedule.standard_morph()
                     self._lambda_schedule.prepend_stage(
                         "restraints_off", self._lambda_schedule.initial()
                     )
@@ -1032,9 +1033,6 @@ def lambda_schedule(self, lambda_schedule):
                         lever="morse_soft",
                         equation=1 - self._lambda_schedule.lam(),
                     )
-                    self._lambda_schedule.set_equation(
-                        stage="restraints_off", lever="morse_hard", equation=0
-                    )
                     self._lambda_schedule.set_equation(
                         stage="restraints_off",
                         lever="bond_k",
@@ -1073,6 +1071,7 @@ def lambda_schedule(self, lambda_schedule):
                         * self._lambda_schedule.initial()
                         + self._lambda_schedule.lam() * self._lambda_schedule.final(),
                     )
+
                     self._lambda_schedule.prepend_stage(
                         "potential_swap", self._lambda_schedule.initial()
                     )
@@ -1120,6 +1119,7 @@ def lambda_schedule(self, lambda_schedule):
                         lever="torsion_phase",
                         equation=self._lambda_schedule.initial(),
                     )
+
                     self._lambda_schedule.set_equation(
                         stage="morph", lever="morse_hard", equation=0
                     )
@@ -1157,6 +1157,7 @@ def lambda_schedule(self, lambda_schedule):
                         equation=self._lambda_schedule.final(),
                     )
                 elif lambda_schedule == "reverse_ring_break_morph":
+                    self._lambda_schedule = _LambdaSchedule.standard_morph()
                     self._lambda_schedule.set_equation(
                         stage="morph", lever="morse_hard", equation=0
                     )
@@ -1291,6 +1292,7 @@ def lambda_schedule(self, lambda_schedule):
                         lever="torsion_phase",
                         equation=self._lambda_schedule.final(),
                     )
+
                 else:
                     try:
                         self._lambda_schedule = self._from_hex(lambda_schedule)