Lazy import in StructureData.

mikibonacci · Sep 9, 2024 · be46122 · be46122
1 parent 090c740
commit be46122
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/src/aiida/orm/nodes/data/structure.py b/src/aiida/orm/nodes/data/structure.py
@@ -22,8 +22,6 @@
 
 from .data import Data
 
-from aiida_atomistic import StructureDataMutable as AtomisticStructureDataMutable
-
 __all__ = ('StructureData', 'Kind', 'Site')
 
 # Threshold used to check if the mass of two different Site objects is the same.
@@ -775,6 +773,7 @@ def to_atomistic(self):
         """
         Returns the atomistic StructureData version of the orm.StructureData one.
         """
+        from aiida_atomistic import StructureDataMutable as AtomisticStructureDataMutable
         atomistic = AtomisticStructureDataMutable(pbc=self.pbc, cell=self.cell)
         for site in self.sites:
             atomistic.add_atom(