TYP: allow for xlim/ylim passed as single tuple

ksunden · ksunden · commit 27976610eea8 · 2023-04-17T15:20:09.000-05:00
Found by running type checker over plotnine/xarray

Also applies to rlim of polar and clim on colormaps
diff --git a/lib/matplotlib/axes/_base.pyi b/lib/matplotlib/axes/_base.pyi
@@ -312,7 +312,7 @@ class _AxesBase(martist.Artist):
     def get_xlim(self) -> tuple[float, float]: ...
     def set_xlim(
         self,
-        left: float | None = ...,
+        left: float | tuple[float, float] | None = ...,
         right: float | None = ...,
         *,
         emit: bool = ...,
@@ -338,7 +338,7 @@ class _AxesBase(martist.Artist):
     def get_ylim(self) -> tuple[float, float]: ...
     def set_ylim(
         self,
-        bottom: float | None = ...,
+        bottom: float | tuple[float, float] | None = ...,
         top: float | None = ...,
         *,
         emit: bool = ...,
diff --git a/lib/matplotlib/cm.pyi b/lib/matplotlib/cm.pyi
@@ -41,7 +41,7 @@ class ScalarMappable:
     def get_array(self) -> np.ndarray | None: ...
     def get_cmap(self) -> colors.Colormap: ...
     def get_clim(self) -> tuple[float, float]: ...
-    def set_clim(self, vmin: float | None = ..., vmax: float | None = ...) -> None: ...
+    def set_clim(self, vmin: float | tuple[float, float] | None = ..., vmax: float | None = ...) -> None: ...
     def get_alpha(self) -> float | None: ...
     def set_cmap(self, cmap: str | colors.Colormap) -> None: ...
     @property
diff --git a/lib/matplotlib/projections/polar.pyi b/lib/matplotlib/projections/polar.pyi
@@ -159,7 +159,7 @@ class PolarAxes(Axes):
     def get_rsign(self) -> float: ...
     def set_rlim(
         self,
-        bottom: float | None = ...,
+        bottom: float | tuple[float, float] | None = ...,
         top: float | None = ...,
         *,
         emit: bool = ...,
