mypy

Author: tsmathis

mp_api/client/core/settings.py

@@ -9,8 +9,10 @@
 
 from mp_api.client import __file__ as root_dir
 
-DEFAULT_THERMOTYPE = ThermoType("GGA_GGA+U_R2SCAN")
-DEFAULT_THERMOTYPE_CRITERIA = {"thermo_types": [DEFAULT_THERMOTYPE.value]}
+DEFAULT_THERMOTYPE: ThermoType = ThermoType("GGA_GGA+U_R2SCAN")
+DEFAULT_THERMOTYPE_CRITERIA: dict[str, list[str]] = {
+    "thermo_types": [DEFAULT_THERMOTYPE.value]
+}
 
 PMG_SETTINGS = _load_pmg_settings()
 _NUM_PARALLEL_REQUESTS = min(PMG_SETTINGS.get("MPRESTER_NUM_PARALLEL_REQUESTS", 4), 4)

mp_api/client/mprester.py

@@ -1590,7 +1590,7 @@ def _get_cohesive_energy(
     def get_stability(
         self,
         entries: list[ComputedEntry | ComputedStructureEntry | PDEntry],
-        thermo_type: str | ThermoType = DEFAULT_THERMOTYPE,
+        thermo_type: ThermoType | str = DEFAULT_THERMOTYPE,
     ) -> list[dict[str, Any]] | None:
         """Get the energy above hull of a list of entries.