Use lattice stub

emmet-core/emmet/stubs/__init__.py

@@ -5,13 +5,15 @@
 """
 
 from pymatgen.analysis.xas.spectrum import XAS
-from pymatgen.core.structure import Composition, Structure
+from pymatgen.core.structure import Composition, Structure, Lattice
 from pymatgen.entries.computed_entries import ComputedEntry
 
 from emmet.stubs.math import Matrix3D, Vector3D
 from emmet.stubs.misc import Composition as StubComposition
 from emmet.stubs.misc import ComputedEntry as StubComputedEntry
 from emmet.stubs.structure import Structure as StubStructure
+from emmet.stubs.structure import Lattice as StubLattice
+
 from emmet.stubs.utils import patch_msonable, use_model
 
 """
@@ -20,6 +22,7 @@
 in as Stubbed classes to prevent name clashing
 """
 use_model(Structure, StubStructure)
+use_model(Lattice, StubLattice)
 use_model(Composition, StubComposition, add_monty=False)
 use_model(ComputedEntry, StubComputedEntry)