Do not compare types, use isinstance()

deepmodeling · Dec 14, 2023 · 37d811b · 37d811b
1 parent 9076806
commit 37d811b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dpti/hti.py b/dpti/hti.py
@@ -390,7 +390,7 @@ def _gen_lammps_input(
     ret += "thermo          ${THERMO_FREQ}\n"
     ret += "compute         allmsd all msd\n"
     if 1 - lamb != 0:
-        if type(m_spring_k) is not list:
+        if not isinstance(m_spring_k, list):
             if switch == "three-step":
                 ret += "thermo_style    custom step ke pe etotal enthalpy temp press vol f_l_spring c_e_diff[1] c_allmsd[*]\n"
             else: