DOI client update

src/mp_api/routes/dois/client.py

@@ -1,26 +1,8 @@
-from mp_api.core.client import BaseRester, MPRestError
+from mp_api.core.client import BaseRester
 from mp_api.routes.dois.models import DOIDoc
 
 
 class DOIRester(BaseRester):
 
     suffix = "doi"
     document_model = DOIDoc  # type: ignore
-
-    def get_eos_from_material_id(self, material_id: str):
-        """
-        Get DOI and reference data for a given Materials Project ID.
-
-        Arguments:
-            material_id (str): Materials project ID
-
-        Returns:
-            results (Dict): Dictionary containing DOI and reference data of state data.
-        """
-
-        result = self._make_request("{}/?all_fields=true".format(material_id))
-
-        if len(result.get("data", [])) > 0:
-            return result
-        else:
-            raise MPRestError("No document found")

src/mp_api/routes/dois/client.pyi

@@ -0,0 +1,13 @@
+from typing import List, Optional
+from mp_api.routes.dois.models import DOIDoc
+
+
+class DOIRester:
+
+    def get_document_by_id(
+        self,
+        document_id: str,
+        fields: Optional[List[str]] = None,
+        monty_decode: bool = True,
+        version: Optional[str] = None,
+    ) -> DOIDoc: ...