Add document model

materialsproject · Dec 2, 2020 · 3e76181 · 3e76181
1 parent 8a73504
commit 3e76181
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/mp_api/molecules/client.py b/src/mp_api/molecules/client.py
@@ -5,11 +5,12 @@
 from pymatgen.core.periodic_table import Element
 
 from mp_api.core.client import BaseRester, MPRestError
-
+from mp_api.molecules.models import MoleculesDoc
 
 class MoleculesRester(BaseRester):
 
     suffix = "molecules"
+    document_model = MoleculesDoc
 
     def get_molecule_from_molecule_id(self, molecule_id: str):
         """