TYP: annotation of __init__ return type (PEP 484) (pandas/plotting) (pandas-dev#46283)

Author: nafarya

pandas/plotting/_core.py

@@ -783,7 +783,7 @@ class PlotAccessor(PandasObject):
     _kind_aliases = {"density": "kde"}
     _all_kinds = _common_kinds + _series_kinds + _dataframe_kinds
 
-    def __init__(self, data):
+    def __init__(self, data) -> None:
         self._parent = data
 
     @staticmethod

pandas/plotting/_matplotlib/boxplot.py

@@ -44,7 +44,7 @@ class BP(NamedTuple):
         ax: Axes
         lines: dict[str, list[Line2D]]
 
-    def __init__(self, data, return_type="axes", **kwargs):
+    def __init__(self, data, return_type="axes", **kwargs) -> None:
         # Do not call LinePlot.__init__ which may fill nan
         if return_type not in self._valid_return_types:
             raise ValueError("return_type must be {None, 'axes', 'dict', 'both'}")

pandas/plotting/_matplotlib/converter.py

@@ -184,7 +184,7 @@ def default_units(x, axis) -> str:
 
 # time formatter
 class TimeFormatter(Formatter):
-    def __init__(self, locs):
+    def __init__(self, locs) -> None:
         self.locs = locs
 
     def __call__(self, x, pos=0) -> str:
@@ -339,7 +339,7 @@ def axisinfo(unit: tzinfo | None, axis) -> units.AxisInfo:
 
 
 class PandasAutoDateFormatter(dates.AutoDateFormatter):
-    def __init__(self, locator, tz=None, defaultfmt="%Y-%m-%d"):
+    def __init__(self, locator, tz=None, defaultfmt="%Y-%m-%d") -> None:
         dates.AutoDateFormatter.__init__(self, locator, tz, defaultfmt)
 
 
@@ -371,7 +371,7 @@ class MilliSecondLocator(dates.DateLocator):
 
     UNIT = 1.0 / (24 * 3600 * 1000)
 
-    def __init__(self, tz):
+    def __init__(self, tz) -> None:
         dates.DateLocator.__init__(self, tz)
         self._interval = 1.0
 
@@ -941,7 +941,7 @@ def __init__(
         month=1,
         day=1,
         plot_obj=None,
-    ):
+    ) -> None:
         freq = to_offset(freq)
         self.freq = freq
         self.base = base
@@ -1025,7 +1025,7 @@ def __init__(
         minor_locator: bool = False,
         dynamic_mode: bool = True,
         plot_obj=None,
-    ):
+    ) -> None:
         freq = to_offset(freq)
         self.format = None
         self.freq = freq

pandas/plotting/_matplotlib/core.py

@@ -128,7 +128,7 @@ def __init__(
         include_bool=False,
         column: IndexLabel | None = None,
         **kwds,
-    ):
+    ) -> None:
 
         import matplotlib.pyplot as plt
 
@@ -984,7 +984,7 @@ class PlanePlot(MPLPlot):
 
     _layout_type = "single"
 
-    def __init__(self, data, x, y, **kwargs):
+    def __init__(self, data, x, y, **kwargs) -> None:
         MPLPlot.__init__(self, data, **kwargs)
         if x is None or y is None:
             raise ValueError(self._kind + " requires an x and y column")
@@ -1037,7 +1037,7 @@ def _plot_colorbar(self, ax: Axes, **kwds):
 class ScatterPlot(PlanePlot):
     _kind = "scatter"
 
-    def __init__(self, data, x, y, s=None, c=None, **kwargs):
+    def __init__(self, data, x, y, s=None, c=None, **kwargs) -> None:
         if s is None:
             # hide the matplotlib default for size, in case we want to change
             # the handling of this argument later
@@ -1125,7 +1125,7 @@ def _make_plot(self):
 class HexBinPlot(PlanePlot):
     _kind = "hexbin"
 
-    def __init__(self, data, x, y, C=None, **kwargs):
+    def __init__(self, data, x, y, C=None, **kwargs) -> None:
         super().__init__(data, x, y, **kwargs)
         if is_integer(C) and not self.data.columns.holds_integer():
             C = self.data.columns[C]
@@ -1157,7 +1157,7 @@ class LinePlot(MPLPlot):
     _default_rot = 0
     orientation = "vertical"
 
-    def __init__(self, data, **kwargs):
+    def __init__(self, data, **kwargs) -> None:
         from pandas.plotting import plot_params
 
         MPLPlot.__init__(self, data, **kwargs)
@@ -1349,7 +1349,7 @@ def get_label(i):
 class AreaPlot(LinePlot):
     _kind = "area"
 
-    def __init__(self, data, **kwargs):
+    def __init__(self, data, **kwargs) -> None:
         kwargs.setdefault("stacked", True)
         data = data.fillna(value=0)
         LinePlot.__init__(self, data, **kwargs)
@@ -1424,7 +1424,7 @@ class BarPlot(MPLPlot):
     _default_rot = 90
     orientation = "vertical"
 
-    def __init__(self, data, **kwargs):
+    def __init__(self, data, **kwargs) -> None:
         # we have to treat a series differently than a
         # 1-column DataFrame w.r.t. color handling
         self._is_series = isinstance(data, ABCSeries)
@@ -1606,7 +1606,7 @@ class PiePlot(MPLPlot):
     _kind = "pie"
     _layout_type = "horizontal"
 
-    def __init__(self, data, kind=None, **kwargs):
+    def __init__(self, data, kind=None, **kwargs) -> None:
         data = data.fillna(value=0)
         if (data < 0).any().any():
             raise ValueError(f"{self._kind} plot doesn't allow negative values")

pandas/plotting/_matplotlib/hist.py

@@ -42,7 +42,7 @@
 class HistPlot(LinePlot):
     _kind = "hist"
 
-    def __init__(self, data, bins=10, bottom=0, **kwargs):
+    def __init__(self, data, bins=10, bottom=0, **kwargs) -> None:
         self.bins = bins  # use mpl default
         self.bottom = bottom
         # Do not call LinePlot.__init__ which may fill nan
@@ -170,7 +170,7 @@ class KdePlot(HistPlot):
     _kind = "kde"
     orientation = "vertical"
 
-    def __init__(self, data, bw_method=None, ind=None, **kwargs):
+    def __init__(self, data, bw_method=None, ind=None, **kwargs) -> None:
         MPLPlot.__init__(self, data, **kwargs)
         self.bw_method = bw_method
         self.ind = ind

pandas/plotting/_misc.py

@@ -521,7 +521,7 @@ class _Options(dict):
     _ALIASES = {"x_compat": "xaxis.compat"}
     _DEFAULT_KEYS = ["xaxis.compat"]
 
-    def __init__(self, deprecated=False):
+    def __init__(self, deprecated=False) -> None:
         self._deprecated = deprecated
         super().__setitem__("xaxis.compat", False)