From 0dadef391c9d5d6763e7341615ec1a61b6fa2a11 Mon Sep 17 00:00:00 2001
From: Thang Nguyen <46436648+thangckt@users.noreply.github.com>
Date: Mon, 20 May 2024 08:25:55 +0900
Subject: [PATCH] Update ase.py

---
 dpdata/plugins/ase.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/dpdata/plugins/ase.py b/dpdata/plugins/ase.py
index 33f1c5ee..01c8483b 100644
--- 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")