From e00011f1e6553377ffca007151d87d194ae2c2ec Mon Sep 17 00:00:00 2001
From: Jason Munro <jason.munro@gmail.com>
Date: Tue, 19 Jul 2022 16:13:44 -0700
Subject: [PATCH] Build thermo with other entry types (#466)

---
 emmet-builders/emmet/builders/vasp/thermo.py | 22 +++++++-------------
 1 file changed, 8 insertions(+), 14 deletions(-)

diff --git a/emmet-builders/emmet/builders/vasp/thermo.py b/emmet-builders/emmet/builders/vasp/thermo.py
index 6f431debd4..4da27051b4 100644
--- a/emmet-builders/emmet/builders/vasp/thermo.py
+++ b/emmet-builders/emmet/builders/vasp/thermo.py
@@ -276,11 +276,6 @@ def get_entries(self, chemsys: str) -> List[Dict]:
         new_q["chemsys"] = {"$in": list(query_chemsys)}
         new_q["deprecated"] = False
 
-        # only build with materials that have a GGA or GGA+U entry
-        new_q["$or"] = [
-            {"entries.GGA": {"$exists": True}},
-            {"entries.GGA+U": {"$exists": True}},
-        ]
         materials_docs = list(
             self.materials.query(
                 criteria=new_q, properties=["material_id", "entries", "deprecated"]
@@ -306,17 +301,16 @@ def get_entries(self, chemsys: str) -> List[Dict]:
             f"Got {len(materials_docs)} entries from DB for {len(query_chemsys)} sub-chemsys for {chemsys}"
         )
 
-        # Convert GGA and GGA+U entries into ComputedEntries and store
+        # Convert entries into ComputedEntries and store
         for doc in materials_docs:
             for r_type, entry_dict in doc.get("entries", {}).items():
-                if r_type == "GGA" or r_type == "GGA+U":
-                    entry_dict["data"]["oxidation_states"] = oxi_states_data.get(
-                        entry_dict["entry_id"], {}
-                    )
-                    entry_dict["data"]["run_type"] = r_type
-                    elsyms = sorted(set([el for el in entry_dict["composition"]]))
-                    self._entries_cache["-".join(elsyms)].append(entry_dict)
-                    all_entries.append(entry_dict)
+                entry_dict["data"]["oxidation_states"] = oxi_states_data.get(
+                    entry_dict["entry_id"], {}
+                )
+                entry_dict["data"]["run_type"] = r_type
+                elsyms = sorted(set([el for el in entry_dict["composition"]]))
+                self._entries_cache["-".join(elsyms)].append(entry_dict)
+                all_entries.append(entry_dict)
 
         self.logger.info(f"Total entries in {chemsys} : {len(all_entries)}")