TYP: Add several missing return type annotations

Author: QuLogic

lib/matplotlib/__init__.pyi

@@ -72,7 +72,7 @@ class RcParams(dict[str, Any]):
     def __getitem__(self, key: str) -> Any: ...
     def __iter__(self) -> Generator[str, None, None]: ...
     def __len__(self) -> int: ...
-    def find_all(self, pattern: str): ...
+    def find_all(self, pattern: str) -> RcParams: ...
     def copy(self) -> RcParams: ...
 
 def rc_params(fail_on_error: bool = ...) -> RcParams: ...
@@ -96,7 +96,7 @@ def rc_file(
 @contextlib.contextmanager
 def rc_context(
     rc: dict[str, Any] | None = ..., fname: str | Path | os.PathLike | None = ...
-): ...
+) -> Generator[None, None, None]: ...
 def use(backend: str, *, force: bool = ...) -> None: ...
 def get_backend() -> str: ...
 def interactive(b: bool) -> None: ...

lib/matplotlib/_api/__init__.pyi

@@ -25,6 +25,7 @@ class classproperty(Any):
         fdel: None = ...,
         doc: str | None = None,
     ): ...
+    # Replace return with Self when py3.9 is dropped
     @overload
     def __get__(self, instance: None, owner: None) -> classproperty: ...
     @overload

lib/matplotlib/_enums.pyi

@@ -1,7 +1,7 @@
 from enum import Enum
 
 class _AutoStringNameEnum(Enum):
-    def __hash__(self): ...
+    def __hash__(self) -> int: ...
 
 class JoinStyle(str, _AutoStringNameEnum):
     miter: str

lib/matplotlib/animation.pyi

@@ -41,7 +41,7 @@ class AbstractMovieWriter(abc.ABC, metaclass=abc.ABCMeta):
     dpi: float
 
     @abc.abstractmethod
-    def setup(self, fig: Figure, outfile: str | Path, dpi: float | None = ...): ...
+    def setup(self, fig: Figure, outfile: str | Path, dpi: float | None = ...) -> None: ...
     @property
     def frame_size(self) -> tuple[int, int]: ...
     @abc.abstractmethod
@@ -65,7 +65,7 @@ class MovieWriter(AbstractMovieWriter):
         extra_args: list[str] | None = ...,
         metadata: dict[str, str] | None = ...,
     ) -> None: ...
-    def setup(self, fig: Figure, outfile: str | Path, dpi: float | None = ...): ...
+    def setup(self, fig: Figure, outfile: str | Path, dpi: float | None = ...) -> None: ...
     def grab_frame(self, **savefig_kwargs) -> None: ...
     def finish(self) -> None: ...
     @classmethod

lib/matplotlib/artist.pyi

@@ -108,9 +108,9 @@ class Artist:
     def set_clip_on(self, b: bool) -> None: ...
     def get_rasterized(self) -> bool: ...
     def set_rasterized(self, rasterized: bool) -> None: ...
-    def get_agg_filter(self): ...
+    def get_agg_filter(self) -> Callable[[ArrayLike, float], tuple[np.ndarray, float, float]] | None: ...
     def set_agg_filter(
-        self, filter_func: Callable[[ArrayLike, float], tuple[np.ndarray, float, float]]
+        self, filter_func: Callable[[ArrayLike, float], tuple[np.ndarray, float, float]] | None
     ) -> None: ...
     def draw(self, renderer: RendererBase) -> None: ...
     def set_alpha(self, alpha: float | None) -> None: ...
@@ -125,9 +125,9 @@ class Artist:
     def sticky_edges(self) -> _XYPair: ...
     def update_from(self, other: Artist) -> None: ...
     def properties(self) -> dict[str, Any]: ...
-    def update(self, props: dict[str, Any]) -> Any: ...
-    def _internal_update(self, kwargs): ...
-    def set(self, **kwargs: Any): ...
+    def update(self, props: dict[str, Any]) -> list[Any]: ...
+    def _internal_update(self, kwargs: Any) -> list[Any]: ...
+    def set(self, **kwargs: Any) -> list[Any]: ...
     def findobj(
         self,
         match: None | Callable[[Artist], bool] | type[Artist] = ...,
@@ -177,5 +177,5 @@ def getp(obj: Artist, property: str | None = ...) -> Any: ...
 
 get = getp
 
-def setp(obj: Artist, *args, file: TextIO | None = ..., **kwargs): ...
+def setp(obj: Artist, *args, file: TextIO | None = ..., **kwargs) -> list[Any] | None: ...
 def kwdoc(artist: Artist | type[Artist] | Iterable[Artist | type[Artist]]) -> str: ...

lib/matplotlib/axes/__init__.pyi

@@ -1,10 +1,13 @@
-from ._axes import *
+from typing import TypeVar
 
+from ._axes import *
 from ._axes import Axes as Subplot
 
+_T = TypeVar("_T")
+
 class _SubplotBaseMeta(type):
     def __instancecheck__(self, obj) -> bool: ...
 
 class SubplotBase(metaclass=_SubplotBaseMeta): ...
 
-def subplot_class_factory(cls): ...
+def subplot_class_factory(cls: type[_T]) -> type[_T]: ...

lib/matplotlib/axes/_secondary_axes.pyi

@@ -36,10 +36,7 @@ class SecondaryAxis(_AxesBase):
     ) -> list[Tick]: ...
     def set_functions(
         self,
-        functions: tuple[
-            Callable[[ArrayLike], ArrayLike], Callable[[ArrayLike], ArrayLike]
-        ]
-        | Transform,
-    ): ...
+        functions: tuple[Callable[[ArrayLike], ArrayLike], Callable[[ArrayLike], ArrayLike]] | Transform,
+    ) -> None: ...
     def set_aspect(self, *args, **kwargs) -> None: ...
     def set_color(self, color: ColorType) -> None: ...

lib/matplotlib/axis.py

@@ -572,7 +572,7 @@ class _LazyTickList:
     def __init__(self, major):
         self._major = major
 
-    def __get__(self, instance, cls):
+    def __get__(self, instance, owner):
         if instance is None:
             return self
         else:

lib/matplotlib/axis.pyi

@@ -1,3 +1,10 @@
+from collections.abc import Callable, Iterable, Sequence
+import datetime
+from typing import Any, Literal, overload
+
+import numpy as np
+from numpy.typing import ArrayLike
+
 import matplotlib.artist as martist
 from matplotlib import cbook
 from matplotlib.axes import Axes
@@ -6,12 +13,6 @@ from matplotlib.lines import Line2D
 from matplotlib.text import Text
 from matplotlib.ticker import Locator, Formatter
 from matplotlib.transforms import Transform, Bbox
-
-import datetime
-from collections.abc import Callable, Iterable, Sequence
-from typing import Any, Literal
-import numpy as np
-from numpy.typing import ArrayLike
 from matplotlib.typing import ColorType
 
 
@@ -92,7 +93,11 @@ class Ticker:
 
 class _LazyTickList:
     def __init__(self, major: bool) -> None: ...
-    def __get__(self, instance: Axis, cls: type): ...
+    # Replace return with Self when py3.9 is dropped
+    @overload
+    def __get__(self, instance: None, owner: None) -> _LazyTickList: ...
+    @overload
+    def __get__(self, instance: Axis, owner: type[Axis]) -> list[Tick]: ...
 
 class Axis(martist.Artist):
     OFFSETTEXTPAD: int

lib/matplotlib/backend_bases.pyi

@@ -76,7 +76,7 @@ class RendererBase:
         facecolors: Sequence[ColorType],
         antialiased: bool,
         edgecolors: Sequence[ColorType] | ColorType | None,
-    ): ...
+    ) -> None: ...
     def draw_gouraud_triangle(
         self,
         gc: GraphicsContextBase,
@@ -321,7 +321,7 @@ class FigureCanvasBase:
     @property
     def scroll_pick_id(self) -> int: ...
     @classmethod
-    def new_manager(cls, figure: Figure, num: int | str): ...
+    def new_manager(cls, figure: Figure, num: int | str) -> FigureManagerBase: ...
     def is_saving(self) -> bool: ...
     def blit(self, bbox: BboxBase | None = ...) -> None: ...
     def inaxes(self, xy: tuple[float, float]) -> Axes | None: ...
@@ -351,7 +351,7 @@ class FigureCanvasBase:
         bbox_extra_artists: list[Artist] | None = ...,
         backend: str | None = ...,
         **kwargs
-    ): ...
+    ) -> Any: ...
     @classmethod
     def get_default_filetype(cls) -> str: ...
     def get_default_filename(self) -> str: ...
@@ -363,7 +363,7 @@ class FigureCanvasBase:
         self,
         interval: int | None = ...,
         callbacks: list[tuple[Callable, tuple, dict[str, Any]]] | None = ...,
-    ): ...
+    ) -> TimerBase: ...
     def flush_events(self) -> None: ...
     def start_event_loop(self, timeout: float = ...) -> None: ...
     def stop_event_loop(self) -> None: ...
@@ -372,7 +372,7 @@ def key_press_handler(
     event: KeyEvent,
     canvas: FigureCanvasBase | None = ...,
     toolbar: NavigationToolbar2 | None = ...,
-): ...
+) -> None: ...
 def button_press_handler(
     event: MouseEvent,
     canvas: FigureCanvasBase | None = ...,
@@ -392,7 +392,7 @@ class FigureManagerBase:
     @classmethod
     def create_with_canvas(
         cls, canvas_class: type[FigureCanvasBase], figure: Figure, num: int | str
-    ): ...
+    ) -> FigureManagerBase: ...
     @classmethod
     def start_main_loop(cls) -> None: ...
     @classmethod
@@ -476,9 +476,9 @@ class _Backend:
     FigureManager: type[FigureManagerBase]
     mainloop: None | Callable[[], Any]
     @classmethod
-    def new_figure_manager(cls, num: int | str, *args, **kwargs): ...
+    def new_figure_manager(cls, num: int | str, *args, **kwargs) -> FigureManagerBase: ...
     @classmethod
-    def new_figure_manager_given_figure(cls, num: int | str, figure: Figure): ...
+    def new_figure_manager_given_figure(cls, num: int | str, figure: Figure) -> FigureManagerBase: ...
     @classmethod
     def draw_if_interactive(cls) -> None: ...
     @classmethod
@@ -487,4 +487,4 @@ class _Backend:
     def export(cls) -> type[_Backend]: ...
 
 class ShowBase(_Backend):
-    def __call__(self, block: bool | None = ...): ...
+    def __call__(self, block: bool | None = ...) -> None: ...