Fix summary to work with new thermo doc (#492)

materialsproject · Aug 2, 2022 · e62a1f6 · e62a1f6
1 parent 2e2031c
commit e62a1f6
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/emmet-builders/emmet/builders/materials/summary.py b/emmet-builders/emmet/builders/materials/summary.py
@@ -116,7 +116,9 @@ def get_items(self):
 
             data = {
                 HasProps.materials.value: materials_doc,
-                HasProps.thermo.value: self.thermo.query_one({self.thermo.key: entry}),
+                HasProps.thermo.value: self.thermo.query_one(
+                    {self.materials.key: entry, "thermo_type": "GGA_GGA+U"}
+                ),
                 HasProps.xas.value: list(self.xas.query({self.xas.key: entry})),
                 HasProps.grain_boundaries.value: list(
                     self.grain_boundaries.query({self.grain_boundaries.key: entry})
@@ -141,8 +143,7 @@ def get_items(self):
                 ),
                 HasProps.insertion_electrodes.value: list(
                     self.insertion_electrodes.query(
-                        {"material_ids": entry},
-                        [self.insertion_electrodes.key],
+                        {"material_ids": entry}, [self.insertion_electrodes.key],
                     )
                 ),
                 HasProps.surface_properties.value: self.surfaces.query_one(