Add Literal-based LogLevel type alias and use it in set_loglevel for stricter typing.

nrnavaneet · nrnavaneet · commit 0555206d4793 · 2025-07-13T04:57:06.000+05:30
diff --git a/lib/matplotlib/__init__.pyi b/lib/matplotlib/__init__.pyi
@@ -40,6 +40,8 @@ from packaging.version import Version
 
 from matplotlib._api import MatplotlibDeprecationWarning
 from typing import Any, Literal, NamedTuple, overload
+from matplotlib.typing import LogLevel
+
 
 class _VersionInfo(NamedTuple):
     major: int
@@ -52,7 +54,7 @@ __bibtex__: str
 __version__: str
 __version_info__: _VersionInfo
 
-def set_loglevel(level: str) -> None: ...
+def set_loglevel(level: LogLevel) -> None: ...
 
 class _ExecInfo(NamedTuple):
     executable: str
diff --git a/lib/matplotlib/pyplot.py b/lib/matplotlib/pyplot.py
@@ -144,6 +144,7 @@
         MouseEventType,
         PickEventType,
         ResizeEventType,
+        LogLevel
     )
     from matplotlib.widgets import SubplotTool
 
@@ -351,7 +352,7 @@ def uninstall_repl_displayhook() -> None:
 
 # Ensure this appears in the pyplot docs.
 @_copy_docstring_and_deprecators(matplotlib.set_loglevel)
-def set_loglevel(level: str) -> None:
+def set_loglevel(level: LogLevel) -> None:
     return matplotlib.set_loglevel(level)
 
 
diff --git a/lib/matplotlib/typing.py b/lib/matplotlib/typing.py
@@ -83,6 +83,9 @@
 CapStyleType: TypeAlias = CapStyle | Literal["butt", "projecting", "round"]
 """Line cap styles. See :doc:`/gallery/lines_bars_and_markers/capstyle`."""
 
+LogLevel: TypeAlias = Literal["notset", "debug", "info", "warning", "error", "critical"]
+"""Literal type for valid logging levels accepted by `set_loglevel()`."""
+
 CoordsBaseType = Union[
     str,
     Artist,
