Remove sort parameters from resters (#864)

* Remove sort params from resters

* Remove 404 from client error

* Linting

Author: Jason Munro

mp_api/client/core/client.py

@@ -817,7 +817,7 @@ def _submit_request_and_process(
                 f"REST query timed out on URL {url}. Try again with a smaller request."
             )
 
-        if response.status_code in [400, 404]:
+        if response.status_code in [400]:
             warnings.warn(
                 f"The server does not support the request made to {response.url}. This may be due to an outdated mp-api package, or a problem with the query."
             )

mp_api/client/core/settings.py

@@ -49,7 +49,6 @@ class MAPIClientSettings(BaseSettings):
             "condition_mixing_media",
             "condition_heating_atmosphere",
             "operations",
-            "_sort_fields",
             "_fields",
         ],
         description="List API query parameters that do not support parallel requests.",

mp_api/client/mprester.py

@@ -657,7 +657,6 @@ def get_entries(
         inc_structure: bool = None,
         property_data: list[str] = None,
         conventional_unit_cell: bool = False,
-        sort_by_e_above_hull: bool = False,
         additional_criteria: dict = None,
     ) -> list[ComputedStructureEntry]:
         """Get a list of ComputedEntries or ComputedStructureEntries corresponding
@@ -688,8 +687,6 @@ def get_entries(
                 input parameters in the 'MPRester.thermo.available_fields' list.
             conventional_unit_cell (bool): Whether to get the standard
                 conventional unit cell
-            sort_by_e_above_hull (bool): Whether to sort the list of entries by
-                e_above_hull in ascending order.
             additional_criteria (dict): Any additional criteria to pass. The keys and values should
                 correspond to proper function inputs to `MPRester.thermo.search`. For instance,
                 if you are only interested in entries on the convex hull, you could pass
@@ -726,19 +723,11 @@ def get_entries(
             else ["entries", "thermo_type"] + property_data
         )
 
-        if sort_by_e_above_hull:
-            docs = self.thermo.search(
-                **input_params,  # type: ignore
-                all_fields=False,
-                fields=fields,
-                sort_fields=["energy_above_hull"],
-            )
-        else:
-            docs = self.thermo.search(
-                **input_params,
-                all_fields=False,
-                fields=fields,  # type: ignore
-            )
+        docs = self.thermo.search(
+            **input_params,
+            all_fields=False,
+            fields=fields,  # type: ignore
+        )
 
         for doc in docs:
             entry_list = (

mp_api/client/routes/_messages.py

@@ -42,7 +42,6 @@ def set_message(
     def get_messages(
         self,
         last_updated: datetime,
-        sort_fields: list[str] | None = None,
         num_chunks: int | None = None,
         chunk_size: int = 1000,
         all_fields: bool = True,
@@ -52,7 +51,6 @@ def get_messages(
 
         Args:
             last_updated (datetime): Datetime to use to query for newer messages
-            sort_fields (List[str]): Fields used to sort results. Prefix with '-' to sort in descending order.
             num_chunks (int): Maximum number of chunks of data to yield. None will yield all possible.
             chunk_size (int): Number of data entries per chunk.
             all_fields (bool): Whether to return all fields in the document. Defaults to True.
@@ -67,11 +65,6 @@ def get_messages(
         """
         query_params = {}
 
-        if sort_fields:
-            query_params.update(
-                {"_sort_fields": ",".join([s.strip() for s in sort_fields])}
-            )
-
         return self._search(
             last_updated=last_updated,
             num_chunks=num_chunks,

mp_api/client/routes/materials/absorption.py

@@ -20,7 +20,6 @@ def search(
         elements: list[str] | None = None,
         exclude_elements: list[str] | None = None,
         formula: list[str] | None = None,
-        sort_fields: list[str] | None = None,
         num_chunks: int | None = None,
         chunk_size: int = 1000,
         all_fields: bool = True,
@@ -37,7 +36,6 @@ def search(
             formula (str, List[str]): A formula including anonymized formula
                 or wild cards (e.g., Fe2O3, ABO3, Si*). A list of chemical formulas can also be passed
                 (e.g., [Fe2O3, ABO3]).
-            sort_fields (List[str]): Fields used to sort results. Prefix with '-' to sort in descending order.
             num_chunks (int): Maximum number of chunks of data to yield. None will yield all possible.
             chunk_size (int): Number of data entries per chunk.
             all_fields (bool): Whether to return all fields in the document. Defaults to True.
@@ -72,11 +70,6 @@ def search(
 
             query_params.update({"material_ids": ",".join(validate_ids(material_ids))})
 
-        if sort_fields:
-            query_params.update(
-                {"_sort_fields": ",".join([s.strip() for s in sort_fields])}
-            )
-
         query_params = {
             entry: query_params[entry]
             for entry in query_params

mp_api/client/routes/materials/alloys.py

@@ -17,7 +17,6 @@ def search(
         self,
         material_ids: str | list[str] | None = None,
         formulae: list[str] | None = None,
-        sort_fields: list[str] | None = None,
         num_chunks: int | None = None,
         chunk_size: int = 1000,
         all_fields: bool = True,
@@ -33,7 +32,6 @@ def search(
         Arguments:
             material_ids (str, List[str]): Search for alloys containing the specified Material IDs
             formulae (List[str]): Search for alloys containing the specified formulae
-            sort_fields (List[str]): Fields used to sort results. Prefix with '-' to sort in descending order.
             num_chunks (int): Maximum number of chunks of data to yield. None will yield all possible.
             chunk_size (int): Number of data entries per chunk.
             all_fields (bool): Whether to return all fields in the document. Defaults to True.
@@ -56,11 +54,6 @@ def search(
 
             query_params.update({"material_ids": ",".join(validate_ids(material_ids))})
 
-        if sort_fields:
-            query_params.update(
-                {"_sort_fields": ",".join([s.strip() for s in sort_fields])}
-            )
-
         return super()._search(
             formulae=formulae,
             num_chunks=num_chunks,

mp_api/client/routes/materials/bonds.py

@@ -32,7 +32,6 @@ def search(
         max_bond_length: tuple[float, float] | None = None,
         mean_bond_length: tuple[float, float] | None = None,
         min_bond_length: tuple[float, float] | None = None,
-        sort_fields: list[str] | None = None,
         num_chunks: int | None = None,
         chunk_size: int = 1000,
         all_fields: bool = True,
@@ -51,7 +50,6 @@ def search(
                 in the structure to consider.
             min_bond_length (Tuple[float,float]): Minimum and maximum value for the minimum bond length
                 in the structure to consider.
-            sort_fields (List[str]): Fields used to sort results. Prefixing with '-' will sort in descending order.
             num_chunks (int): Maximum number of chunks of data to yield. None will yield all possible.
             chunk_size (int): Number of data entries per chunk.
             all_fields (bool): Whether to return all fields in the document. Defaults to True.
@@ -101,11 +99,6 @@ def search(
                 {"coordination_envs_anonymous": ",".join(coordination_envs_anonymous)}
             )
 
-        if sort_fields:
-            query_params.update(
-                {"_sort_fields": ",".join([s.strip() for s in sort_fields])}
-            )
-
         query_params = {
             entry: query_params[entry]
             for entry in query_params

mp_api/client/routes/materials/chemenv.py

@@ -42,7 +42,6 @@ def search(
         num_elements: tuple[int, int] | None = None,
         num_sites: tuple[int, int] | None = None,
         volume: tuple[float, float] | None = None,
-        sort_fields: list[str] | None = None,
         num_chunks: int | None = None,
         chunk_size: int = 1000,
         all_fields: bool = True,
@@ -68,7 +67,6 @@ def search(
             num_elements (Tuple[int,int]): Minimum and maximum number of elements to consider.
             num_sites (Tuple[int,int]): Minimum and maximum number of sites to consider.
             volume (Tuple[float,float]): Minimum and maximum volume to consider.
-            sort_fields (List[str]): Fields used to sort results. Prefix with '-' to sort in descending order.
             num_chunks (int): Maximum number of chunks of data to yield. None will yield all possible.
             chunk_size (int): Number of data entries per chunk.
             all_fields (bool): Whether to return all fields in the document. Defaults to True.
@@ -136,11 +134,6 @@ def search(
 
             query_params.update({"species": ",".join(species)})
 
-        if sort_fields:
-            query_params.update(
-                {"_sort_fields": ",".join([s.strip() for s in sort_fields])}
-            )
-
         query_params = {
             entry: query_params[entry]
             for entry in query_params

mp_api/client/routes/materials/dielectric.py

@@ -31,7 +31,6 @@ def search(
         e_ionic: tuple[float, float] | None = None,
         e_electronic: tuple[float, float] | None = None,
         n: tuple[float, float] | None = None,
-        sort_fields: list[str] | None = None,
         num_chunks: int | None = None,
         chunk_size: int = 1000,
         all_fields: bool = True,
@@ -46,7 +45,6 @@ def search(
             e_ionic (Tuple[float,float]): Minimum and maximum ionic dielectric constant to consider.
             e_electronic (Tuple[float,float]): Minimum and maximum electronic dielectric constant to consider.
             n (Tuple[float,float]): Minimum and maximum refractive index to consider.
-            sort_fields (List[str]): Fields used to sort results. Prefix with '-' to sort in descending order.
             num_chunks (int): Maximum number of chunks of data to yield. None will yield all possible.
             chunk_size (int): Number of data entries per chunk.
             all_fields (bool): Whether to return all fields in the document. Defaults to True.
@@ -81,11 +79,6 @@ def search(
         if n:
             query_params.update({"n_min": n[0], "n_max": n[1]})
 
-        if sort_fields:
-            query_params.update(
-                {"_sort_fields": ",".join([s.strip() for s in sort_fields])}
-            )
-
         query_params = {
             entry: query_params[entry]
             for entry in query_params

mp_api/client/routes/materials/elasticity.py

@@ -35,7 +35,6 @@ def search(
         k_reuss: tuple[float, float] | None = None,
         k_vrh: tuple[float, float] | None = None,
         poisson_ratio: tuple[float, float] | None = None,
-        sort_fields: list[str] | None = None,
         num_chunks: int | None = None,
         chunk_size: int = 1000,
         all_fields: bool = True,
@@ -62,7 +61,6 @@ def search(
                 the Voigt-Reuss-Hill average of the bulk modulus.
             poisson_ratio (Tuple[float,float]): Minimum and maximum value to consider for
                 Poisson's ratio.
-            sort_fields (List[str]): Fields used to sort results. Prefix with '-' to sort in descending order.
             num_chunks (int): Maximum number of chunks of data to yield. None will yield all possible.
             chunk_size (int): Number of data entries per chunk.
             all_fields (bool): Whether to return all fields in the document. Defaults to True.
@@ -111,11 +109,6 @@ def search(
                 {"poisson_min": poisson_ratio[0], "poisson_max": poisson_ratio[1]}
             )
 
-        if sort_fields:
-            query_params.update(
-                {"_sort_fields": ",".join([s.strip() for s in sort_fields])}
-            )
-
         query_params = {
             entry: query_params[entry]
             for entry in query_params