@@ -226,6 +226,14 @@ def __init__(
226226 else :
227227 raise IOError ("SOMD executable doesn't exist: '%s'" % exe )
228228
229+ # Validate torsion modification kwargs.
230+ self ._zero_dummy_dihedrals = kwargs .get ("zero_dummy_dihedrals" , False )
231+ if not isinstance (self ._zero_dummy_dihedrals , bool ):
232+ self ._zero_dummy_dihedrals = False
233+ self ._zero_dummy_impropers = kwargs .get ("zero_dummy_impropers" , False )
234+ if not isinstance (self ._zero_dummy_impropers , bool ):
235+ self ._zero_dummy_impropers = False
236+
229237 # The names of the input files.
230238 self ._rst_file = _os .path .join (str (self ._work_dir ), f"{ name } .rst7" )
231239 self ._top_file = _os .path .join (str (self ._work_dir ), f"{ name } .prm7" )
@@ -322,7 +330,9 @@ def _setup(self):
322330 # to the lambda = 0 state.
323331 pert_mol = _to_pert_file (
324332 pert_mol ,
325- self ._pert_file ,
333+ filename = self ._pert_file ,
334+ zero_dummy_dihedrals = self ._zero_dummy_dihedrals ,
335+ zero_dummy_impropers = self ._zero_dummy_impropers ,
326336 property_map = self ._property_map ,
327337 perturbation_type = self ._protocol .getPerturbationType (),
328338 )
0 commit comments