From c8f4aa4bb201ca2995885d17cbcfd413c5dee2c5 Mon Sep 17 00:00:00 2001 From: Alex Ganose Date: Mon, 26 Oct 2020 13:22:15 -0700 Subject: [PATCH] Use lattice stub --- emmet-core/emmet/stubs/__init__.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/emmet-core/emmet/stubs/__init__.py b/emmet-core/emmet/stubs/__init__.py index a56c4d1036..86b0951678 100644 --- a/emmet-core/emmet/stubs/__init__.py +++ b/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)