add missing import (#116)

lanl · Oct 22, 2024 · 862ed0f · 862ed0f
1 parent 6c6d9d1
commit 862ed0f
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/hippynn/molecular_dynamics/md.py b/hippynn/molecular_dynamics/md.py
@@ -4,6 +4,7 @@
 
 import numpy as np
 import torch
+import ase
 
 from ..tools import progress_bar
 from ..graphs import Predictor