fix import issue

Signed-off-by: Jinzhe Zeng <jinzhe.zeng@rutgers.edu>
deepmodeling · May 17, 2024 · 04e584b · 04e584b
1 parent 69d4fa0
commit 04e584b
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/dpdata/data_type.py b/dpdata/data_type.py
@@ -5,7 +5,6 @@
 
 import numpy as np
 
-from dpdata.bond_order_system import BondOrderSystem
 from dpdata.plugin import Plugin
 
 if TYPE_CHECKING:
@@ -74,8 +73,7 @@ def real_shape(self, system: System) -> tuple[int]:
                 shape.append(system.get_natoms())
             elif ii is Axis.NBONDS:
                 # BondOrderSystem
-                assert isinstance(system, BondOrderSystem)
-                shape.append(system.get_nbonds())
+                shape.append(system.get_nbonds())  # type: ignore
             elif ii == -1:
                 shape.append(AnyInt(-1))
             elif isinstance(ii, int):