change to mprest warning

Author: esoteric-ephemera

mp_api/client/mprester.py

@@ -22,7 +22,7 @@
 from pymatgen.symmetry.analyzer import SpacegroupAnalyzer
 from requests import Session, get
 
-from mp_api.client.core import BaseRester, MPRestError
+from mp_api.client.core import BaseRester, MPRestError, MPRestWarning
 from mp_api.client.core._oxygen_evolution import OxygenEvolution
 from mp_api.client.core.settings import MAPIClientSettings
 from mp_api.client.core.utils import load_json, validate_api_key, validate_ids
@@ -213,7 +213,9 @@ def __init__(
         if "monty_decode" in kwargs:
             warnings.warn(
                 "Ignoring `monty_decode`, as it is no longer a supported option in `mp_api`."
-                "The client by default returns results consistent with `monty_decode=True`."
+                "The client by default returns results consistent with `monty_decode=True`.",
+                stacklevel=2,
+                category=MPRestWarning,
             )
 
         # Check if emmet version of server is compatible

tests/test_mprester.py

@@ -33,7 +33,7 @@
 from pymatgen.io.vasp import Chgcar
 
 from mp_api.client import MPRester
-from mp_api.client.core.exceptions import MPRestError
+from mp_api.client.core import MPRestError, MPRestWarning
 from mp_api.client.core.settings import MAPIClientSettings
 
 from .conftest import requires_api_key
@@ -575,3 +575,7 @@ def test_oxygen_evolution_bad_input(self, mpr):
 
         with pytest.raises(ValueError, match="No available insertion electrode data"):
             _ = mpr.get_oxygen_evolution("mp-2207", "Al")
+
+    def test_monty_decode_warning(self):
+        with pytest.warns(MPRestWarning, match="Ignoring `monty_decode`"):
+            MPRester(monty_decode=False)