Update ase.py

deepmodeling · May 19, 2024 · 0dadef3 · 0dadef3
1 parent c5fb9ba
commit 0dadef3
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/dpdata/plugins/ase.py b/dpdata/plugins/ase.py
@@ -11,6 +11,7 @@
 if TYPE_CHECKING:
     import ase
     from ase.optimize.optimize import Optimizer
+    from ase.io import Trajectory
 
 
 @Format.register("ase/structure")