thermostat: correct allowed thermostat combinations.

espressomd · Dec 3, 2019 · b081309 · b081309
1 parent ab5dfb0
commit b081309
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/python/espressomd/thermostat.pyx b/src/python/espressomd/thermostat.pyx
@@ -205,7 +205,7 @@ cdef class Thermostat:
         lb_lbcoupling_set_gamma(0.0)
         return True
 
-    @AssertThermostatType(THERMO_LANGEVIN)
+    @AssertThermostatType(THERMO_LANGEVIN, THERMO_DPD)
     def set_langevin(self, kT=None, gamma=None, gamma_rotation=None,
                      act_on_virtual=False, seed=None):
         """
@@ -358,7 +358,7 @@ cdef class Thermostat:
             mpi_bcast_parameter(FIELD_LANGEVIN_GAMMA_ROTATION)
         return True
 
-    @AssertThermostatType(THERMO_LB)
+    @AssertThermostatType(THERMO_LB, THERMO_DPD)
     def set_lb(
         self,
         seed=None,