Merge pull request #27217 from ksunden/contour_type

rcomer · web-flow · commit 518b268a5e94 · 2023-10-28T13:35:48.000+01:00
Fix type hints for undeprecated contour APIs
diff --git a/ci/mypy-stubtest-allowlist.txt b/ci/mypy-stubtest-allowlist.txt
@@ -41,12 +41,6 @@ matplotlib.figure.Figure.set_tight_layout
 matplotlib.cm.register_cmap
 matplotlib.cm.unregister_cmap
 
-# 3.8 deprecations
-matplotlib.contour.ContourSet.allkinds
-matplotlib.contour.ContourSet.allsegs
-matplotlib.contour.ContourSet.tcolors
-matplotlib.contour.ContourSet.tlinewidths
-
 # positional-only argument name lacking leading underscores
 matplotlib.axes._base._AxesBase.axis
 
diff --git a/lib/matplotlib/contour.pyi b/lib/matplotlib/contour.pyi
@@ -103,12 +103,17 @@ class ContourSet(ContourLabeler, Collection):
     clip_path: Patch | Path | TransformedPath | TransformedPatchPath | None
     labelTexts: list[Text]
     labelCValues: list[ColorType]
-    allkinds: list[np.ndarray]
-    tcolors: list[tuple[float, float, float, float]]
+    @property
+    def tcolors(self) -> list[tuple[tuple[float, float, float, float]]]: ...
 
     # only for not filled
-    tlinewidths: list[tuple[float]]
+    @property
+    def tlinewidths(self) -> list[tuple[float]]: ...
 
+    @property
+    def allkinds(self) -> list[list[np.ndarray | None]]: ...
+    @property
+    def allsegs(self) -> list[list[np.ndarray]]: ...
     @property
     def alpha(self) -> float | None: ...
     @property
diff --git a/lib/matplotlib/path.pyi b/lib/matplotlib/path.pyi
@@ -30,7 +30,7 @@ class Path:
     @vertices.setter
     def vertices(self, vertices: ArrayLike) -> None: ...
     @property
-    def codes(self) -> ArrayLike: ...
+    def codes(self) -> ArrayLike | None: ...
     @codes.setter
     def codes(self, codes: ArrayLike) -> None: ...
     @property
