add documentation for new md code (#77)

lanl · Jun 6, 2024 · 3f0c605 · 3f0c605
1 parent 41ad80b
commit 3f0c605
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/hippynn/molecular_dynamics/__init__.py b/hippynn/molecular_dynamics/__init__.py
@@ -1,6 +1,5 @@
-from .md import Variable, NullUpdater, VelocityVerlet, LangevinDynamics, MolecularDynamics
-
 """
 Molecular dynamics driver with great flexibility and customizability regarding which quantities which are evolved 
 and what algorithms are used to evolve them. Calls a hippynn `Predictor` on current state during each MD step. 
-"""
+"""
+from .md import Variable, NullUpdater, VelocityVerlet, LangevinDynamics, MolecularDynamics