Update scale.pyi

vagnermcj · vagnermcj · commit 93bf34b48b52 · 2025-05-13T15:26:12.000-03:00
diff --git a/lib/matplotlib/scale.pyi b/lib/matplotlib/scale.pyi
@@ -2,7 +2,7 @@ from matplotlib.axis import Axis
 from matplotlib.transforms import Transform
 
 from collections.abc import Callable, Iterable
-from typing import Literal
+from typing import Literal, Union
 from numpy.typing import ArrayLike
 
 class ScaleBase:
@@ -15,6 +15,7 @@ class ScaleBase:
 
 class LinearScale(ScaleBase):
     name: str
+    def __init__(self: ScaleBase, axis: Union[Axis, None] = None) -> None: ...
 
 class FuncTransform(Transform):
     input_dims: int
@@ -56,12 +57,12 @@ class LogScale(ScaleBase):
     name: str
     subs: Iterable[int] | None
     def __init__(
-        self,
-        axis: Axis | None,
+        self: LogScale,
+        axis: Union[Axis, None] = None,
         *,
-        base: float = ...,
-        subs: Iterable[int] | None = ...,
-        nonpositive: Literal["clip", "mask"] = ...
+        base: float = 10,
+        subs: Union[Iterable[int], None] = None,
+        nonpositive: Union[Literal['clip'], Literal['mask']] = 'clip'
     ) -> None: ...
     @property
     def base(self) -> float: ...
@@ -103,13 +104,13 @@ class SymmetricalLogScale(ScaleBase):
     name: str
     subs: Iterable[int] | None
     def __init__(
-        self,
-        axis: Axis | None,
+        self: SymmetricalLogScale,
+        axis: Union[Axis, None] = None,
         *,
-        base: float = ...,
-        linthresh: float = ...,
-        subs: Iterable[int] | None = ...,
-        linscale: float = ...
+        base: float = 10,
+        linthresh: float = 2,
+        subs: Union[Iterable[int], None] = None,
+        linscale: float = 1
     ) -> None: ...
     @property
     def base(self) -> float: ...
@@ -164,15 +165,16 @@ class LogisticTransform(Transform):
 class LogitScale(ScaleBase):
     name: str
     def __init__(
-        self,
-        axis: Axis | None,
-        nonpositive: Literal["mask", "clip"] = ...,
+        self: LogitScale,
+        axis: Union[Axis, None] = None,
+        nonpositive: Union[Literal['mask'], Literal['clip']] = 'mask',
         *,
-        one_half: str = ...,
-        use_overline: bool = ...
+        one_half: str = '\\frac{1}{2}',
+        use_overline: bool = False
     ) -> None: ...
     def get_transform(self) -> LogitTransform: ...
 
 def get_scale_names() -> list[str]: ...
 def scale_factory(scale: str, axis: Axis, **kwargs) -> ScaleBase: ...
 def register_scale(scale_class: type[ScaleBase]) -> None: ...
+def handle_axis_parameter(init_func: Callable[..., None]) -> Callable[..., None]: ...
