Change model to use pmg Structure, leave Composition as-is

materialsproject · Apr 1, 2021 · e2321cf · e2321cf
1 parent 7657621
commit e2321cf
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/mp_api/search/models.py b/src/mp_api/search/models.py
@@ -1,8 +1,8 @@
 from typing import List, Optional
 from pymatgen.core.periodic_table import Element
+from pymatgen.core.structure import Structure
 
-from mp_api.materials.models import Structure, Composition
-from mp_api.materials.models.doc import SymmetryData
+from mp_api.materials.models.doc import SymmetryData, Composition
 from mp_api.xas.models import Edge, XASType
 from mp_api.gb.models import GBTypeEnum