fixed docstring for md ensemble

CederGroupHub · Mar 1, 2024 · 3719f70 · 3719f70
1 parent 2fb442d
commit 3719f70
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/chgnet/model/dynamics.py b/chgnet/model/dynamics.py
@@ -439,7 +439,7 @@ def __init__(
             model (CHGNet): instance of a CHGNet model or CHGNetCalculator.
                 If set to None, the pretrained CHGNet is loaded.
                 Default = None
-            ensemble (str): choose from 'nve', 'nvt', 'npt', 'npt_berendsen'
+            ensemble (str): choose from 'nve', 'nvt', 'npt'
                 Default = "nvt"
             thermostat (str): Thermostat to use
                 choose from "Nose-Hoover", "Berendsen", "Berendsen_inhomogeneous"