nelements added to general min max filter

materialsproject · Mar 26, 2021 · 55fce08 · 55fce08
1 parent 446967c
commit 55fce08
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/src/mp_api/materials/query_operators.py b/src/mp_api/materials/query_operators.py
@@ -80,6 +80,12 @@ def query(
         nsites_min: Optional[int] = Query(
             None, description="Minimum value for the number of sites",
         ),
+        nelements_max: Optional[float] = Query(
+            None, description="Maximum value for the number of elements.",
+        ),
+        nelements_min: Optional[float] = Query(
+            None, description="Minimum value for the number of elements.",
+        ),
         volume_max: Optional[float] = Query(
             None, description="Maximum value for the cell volume",
         ),
@@ -98,6 +104,7 @@ def query(
 
         entries = {
             "nsites": [nsites_min, nsites_max],
+            "nelements": [nelements_min, nelements_max],
             "volume": [volume_min, volume_max],
             "density": [density_min, density_max],
         }  # type: dict