Chunk with chemsys in structure group builder (#289)

materialsproject · Oct 25, 2021 · e3e072b · e3e072b
1 parent ae6c5a9
commit e3e072b
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/emmet-builders/emmet/builders/materials/electrodes.py b/emmet-builders/emmet/builders/materials/electrodes.py
@@ -112,11 +112,11 @@ def prechunk(self, number_splits: int) -> Iterator[Dict]:
         """
         q = dict(self.query)
 
-        keys = self.sgroups.newer_in(self.materials, criteria=q, exhaustive=True)
+        all_chemsys = self.materials.distinct("chemsys", criteria=q)
 
-        N = ceil(len(keys) / number_splits)
-        for split in grouper(keys, N):
-            yield {"query": {self.materials.key: {"$in": list(split)}}}
+        N = ceil(len(all_chemsys) / number_splits)
+        for split in grouper(all_chemsys, N):
+            yield {"query": {"chemsys": {"$in": list(split)}}}
 
     def get_items(self):
         """