Fix max list

materialsproject · Dec 8, 2023 · 741a14e · 741a14e
1 parent 6a091b9
commit 741a14e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/mp_api/client/core/utils.py b/mp_api/client/core/utils.py
@@ -25,7 +25,7 @@ def validate_ids(id_list: list[str]):
     Returns:
         id_list: Returns original ID list if everything is formatted correctly.
     """
-    if len(id_list) >= MAPIClientSettings().MAX_LIST_LENGTH:
+    if len(id_list) > MAPIClientSettings().MAX_LIST_LENGTH:
         raise ValueError(
             "List of material/molecule IDs provided is too long. Consider removing the ID filter to automatically pull"
             " data for all IDs and filter locally."

diff --git a/tests/test_mprester.py b/tests/test_mprester.py
@@ -316,7 +316,7 @@ def test_large_list(self, mpr):
         mpids = [
             str(doc.material_id)
             for doc in mpr.summary.search(
-                chunk_size=1000, num_chunks=15, fields=["material_id"]
+                chunk_size=1000, num_chunks=10, fields=["material_id"]
             )
         ]
         docs = mpr.summary.search(material_ids=mpids, fields=["material_ids"])