Piezo chunk size and doc string update

materialsproject · May 11, 2021 · 62f97fe · 62f97fe
1 parent d2750cd
commit 62f97fe
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/src/mp_api/routes/piezo/client.py b/src/mp_api/routes/piezo/client.py
@@ -16,7 +16,7 @@ def search_piezoelectric_docs(
         self,
         piezoelectric_modulus: Optional[Tuple[float, float]] = None,
         num_chunks: Optional[int] = None,
-        chunk_size: int = 100,
+        chunk_size: int = 1000,
         all_fields: bool = True,
         fields: Optional[List[str]] = None,
     ):
@@ -32,9 +32,8 @@ def search_piezoelectric_docs(
             fields (List[str]): List of fields in EOSDoc to return data for.
                 Default is material_id only.
 
-        Yields:
-            ([dict]) List of dictionaries containing data for entries defined in 'fields'.
-                Defaults to Materials Project IDs only.
+        Returns:
+            ([PiezoDoc]) List of piezoelectric documents
         """
 
         query_params = defaultdict(dict)  # type: dict