enable reportUnknownVariableType

Author: cmp0xff

.pre-commit-config.yaml

@@ -18,6 +18,6 @@ repos:
     hooks:
     -   id: isort
 -   repo: https://github.com/psf/black
-    rev: 25.9.0
+    rev: 25.11.0
     hooks:
     -   id: black

pandas-stubs/__init__.pyi

@@ -61,16 +61,20 @@ from pandas.core.api import (
     timedelta_range as timedelta_range,
     to_datetime as to_datetime,
     to_numeric as to_numeric,
-    to_timedelta as to_timedelta,
     unique as unique,
-    value_counts as value_counts,
+)
+from pandas.core.api import (
+    to_timedelta as to_timedelta,  # pyright: ignore[reportUnknownVariableType]
+)
+from pandas.core.api import (
+    value_counts as value_counts,  # pyright: ignore[reportUnknownVariableType]
 )
 from pandas.core.arrays.sparse import SparseDtype as SparseDtype
-from pandas.core.computation.api import eval as eval
+from pandas.core.computation.api import (
+    eval as eval,  # pyright: ignore[reportUnknownVariableType]
+)
 from pandas.core.reshape.api import (
     concat as concat,
-    crosstab as crosstab,
-    cut as cut,
     from_dummies as from_dummies,
     get_dummies as get_dummies,
     lreshape as lreshape,
@@ -79,10 +83,18 @@ from pandas.core.reshape.api import (
     merge_asof as merge_asof,
     merge_ordered as merge_ordered,
     pivot as pivot,
-    pivot_table as pivot_table,
     qcut as qcut,
     wide_to_long as wide_to_long,
 )
+from pandas.core.reshape.api import (
+    crosstab as crosstab,  # pyright: ignore[reportUnknownVariableType]
+)
+from pandas.core.reshape.api import (
+    cut as cut,  # pyright: ignore[reportUnknownVariableType]
+)
+from pandas.core.reshape.api import (
+    pivot_table as pivot_table,  # pyright: ignore[reportUnknownVariableType]
+)
 
 from pandas._config import (
     describe_option as describe_option,
@@ -99,27 +111,49 @@ from pandas.io.api import (
     ExcelFile as ExcelFile,
     ExcelWriter as ExcelWriter,
     HDFStore as HDFStore,
-    read_clipboard as read_clipboard,
-    read_csv as read_csv,
-    read_excel as read_excel,
     read_feather as read_feather,
     read_fwf as read_fwf,
     read_hdf as read_hdf,
-    read_html as read_html,
     read_json as read_json,
-    read_orc as read_orc,
-    read_parquet as read_parquet,
     read_pickle as read_pickle,
     read_sas as read_sas,
     read_spss as read_spss,
-    read_sql as read_sql,
-    read_sql_query as read_sql_query,
     read_sql_table as read_sql_table,
     read_stata as read_stata,
-    read_table as read_table,
-    read_xml as read_xml,
 )
-from pandas.io.json._normalize import json_normalize as json_normalize
+from pandas.io.api import (
+    read_clipboard as read_clipboard,  # pyright: ignore[reportUnknownVariableType]
+)
+from pandas.io.api import (
+    read_csv as read_csv,  # pyright: ignore[reportUnknownVariableType]
+)
+from pandas.io.api import (
+    read_excel as read_excel,  # pyright: ignore[reportUnknownVariableType]
+)
+from pandas.io.api import (
+    read_html as read_html,  # pyright: ignore[reportUnknownVariableType]
+)
+from pandas.io.api import (
+    read_orc as read_orc,  # pyright: ignore[reportUnknownVariableType]
+)
+from pandas.io.api import (
+    read_parquet as read_parquet,  # pyright: ignore[reportUnknownVariableType]
+)
+from pandas.io.api import (
+    read_sql as read_sql,  # pyright: ignore[reportUnknownVariableType]
+)
+from pandas.io.api import (
+    read_sql_query as read_sql_query,  # pyright: ignore[reportUnknownVariableType]
+)
+from pandas.io.api import (
+    read_table as read_table,  # pyright: ignore[reportUnknownVariableType]
+)
+from pandas.io.api import (
+    read_xml as read_xml,  # pyright: ignore[reportUnknownVariableType]
+)
+from pandas.io.json._normalize import (
+    json_normalize as json_normalize,  # pyright: ignore[reportUnknownVariableType]
+)
 from pandas.tseries import offsets as offsets
 from pandas.tseries.api import infer_freq as infer_freq
 

pandas-stubs/_typing.pyi

@@ -22,7 +22,6 @@ from typing import (
     SupportsIndex,
     TypeAlias,
     TypedDict,
-    Union,
     overload,
 )
 
@@ -596,18 +595,25 @@ IndexKeyFunc: TypeAlias = Callable[[Index], Index | AnyArrayLike] | None
 
 # types of `func` kwarg for DataFrame.aggregate and Series.aggregate
 # More specific than what is in pandas
-# following Union is here to make it ty compliant https://github.com/astral-sh/ty/issues/591
-AggFuncTypeBase: TypeAlias = Union[Callable, str, np.ufunc]  # noqa: UP007
+AggFuncTypeBase: TypeAlias = (  # pyright: ignore[reportUnknownVariableType]
+    Callable | str | np.ufunc
+)
 AggFuncTypeDictSeries: TypeAlias = Mapping[HashableT, AggFuncTypeBase]
 AggFuncTypeDictFrame: TypeAlias = Mapping[
     HashableT, AggFuncTypeBase | list[AggFuncTypeBase]
 ]
-AggFuncTypeSeriesToFrame: TypeAlias = list[AggFuncTypeBase] | AggFuncTypeDictSeries
-AggFuncTypeFrame: TypeAlias = (
+AggFuncTypeSeriesToFrame: TypeAlias = (  # pyright: ignore[reportUnknownVariableType]
+    list[AggFuncTypeBase] | AggFuncTypeDictSeries
+)
+AggFuncTypeFrame: TypeAlias = (  # pyright: ignore[reportUnknownVariableType]
     AggFuncTypeBase | list[AggFuncTypeBase] | AggFuncTypeDictFrame
 )
-AggFuncTypeDict: TypeAlias = AggFuncTypeDictSeries | AggFuncTypeDictFrame
-AggFuncType: TypeAlias = AggFuncTypeBase | list[AggFuncTypeBase] | AggFuncTypeDict
+AggFuncTypeDict: TypeAlias = (  # pyright: ignore[reportUnknownVariableType]
+    AggFuncTypeDictSeries | AggFuncTypeDictFrame
+)
+AggFuncType: TypeAlias = (  # pyright: ignore[reportUnknownVariableType]
+    AggFuncTypeBase | list[AggFuncTypeBase] | AggFuncTypeDict
+)
 
 # Not used in stubs
 # AggObjType = Union[

pandas-stubs/core/api.pyi

@@ -1,7 +1,9 @@
 from pandas.core.algorithms import (
     factorize as factorize,
     unique as unique,
-    value_counts as value_counts,
+)
+from pandas.core.algorithms import (
+    value_counts as value_counts,  # pyright: ignore[reportUnknownVariableType]
 )
 from pandas.core.arrays import Categorical as Categorical
 from pandas.core.arrays.arrow.dtype import ArrowDtype as ArrowDtype
@@ -51,7 +53,9 @@ from pandas.core.indexing import IndexSlice as IndexSlice
 from pandas.core.series import Series as Series
 from pandas.core.tools.datetimes import to_datetime as to_datetime
 from pandas.core.tools.numeric import to_numeric as to_numeric
-from pandas.core.tools.timedeltas import to_timedelta as to_timedelta
+from pandas.core.tools.timedeltas import (
+    to_timedelta as to_timedelta,  # pyright: ignore[reportUnknownVariableType]
+)
 
 from pandas._libs import (
     NaT as NaT,

pandas-stubs/core/computation/api.pyi

@@ -1 +1,3 @@
-from pandas.core.computation.eval import eval as eval
+from pandas.core.computation.eval import (
+    eval as eval,  # pyright: ignore[reportUnknownVariableType]
+)

pandas-stubs/core/frame.pyi

@@ -57,7 +57,9 @@ from pandas.core.indexing import (
     _LocIndexer,
 )
 from pandas.core.reshape.pivot import (
-    _PivotAggFunc,
+    _PivotAggFunc,  # pyright: ignore[reportUnknownVariableType]
+)
+from pandas.core.reshape.pivot import (
     _PivotTableColumnsTypes,
     _PivotTableIndexTypes,
     _PivotTableValuesTypes,
@@ -81,12 +83,13 @@ from pandas._libs.lib import _NoDefaultDoNotUse
 from pandas._libs.missing import NAType
 from pandas._libs.tslibs import BaseOffset
 from pandas._libs.tslibs.nattype import NaTType
+from pandas._typing import (
+    AggFuncTypeFrame,  # pyright: ignore[reportUnknownVariableType]
+)
 from pandas._typing import (
     S2,
-    AggFuncTypeBase,
     AggFuncTypeDictFrame,
     AggFuncTypeDictSeries,
-    AggFuncTypeFrame,
     AlignJoin,
     AnyAll,
     AnyArrayLike,
@@ -171,6 +174,7 @@ from pandas._typing import (
     np_ndarray_float,
     np_ndarray_num,
 )
+from pandas._typing import AggFuncTypeBase  # pyright: ignore[reportUnknownVariableType]
 
 from pandas.io.formats.style import Styler
 from pandas.plotting import PlotAccessor
@@ -1527,7 +1531,7 @@ class DataFrame(NDFrame, OpsMixin, _GetItemHack):
         axis: Axis = 0,
         **kwargs: Any,
     ) -> Self: ...
-    agg = aggregate
+    agg = aggregate  # pyright: ignore[reportUnknownVariableType]
     def transform(
         self,
         func: AggFuncTypeFrame,

pandas-stubs/core/groupby/generic.pyi

@@ -29,11 +29,12 @@ from pandas.core.series import Series
 from typing_extensions import Self
 
 from pandas._libs.tslibs.timestamps import Timestamp
+from pandas._typing import (
+    AggFuncTypeFrame,  # pyright: ignore[reportUnknownVariableType]
+)
 from pandas._typing import (
     S2,
     S3,
-    AggFuncTypeBase,
-    AggFuncTypeFrame,
     ByT,
     CorrelationMethod,
     Dtype,
@@ -48,6 +49,7 @@ from pandas._typing import (
     WindowingEngineKwargs,
     np_ndarray,
 )
+from pandas._typing import AggFuncTypeBase  # pyright: ignore[reportUnknownVariableType]
 
 AggScalar: TypeAlias = str | Callable[..., Any]
 
@@ -95,7 +97,7 @@ class SeriesGroupBy(GroupBy[Series[S2]], Generic[S2, ByT]):
         engine_kwargs: WindowingEngineKwargs = ...,
         **kwargs: Any,
     ) -> Series: ...
-    agg = aggregate
+    agg = aggregate  # pyright: ignore[reportUnknownVariableType]
     @overload
     def transform(
         self,
@@ -270,7 +272,7 @@ class DataFrameGroupBy(GroupBy[DataFrame], Generic[ByT, _TT]):
         /,
         **kwargs: Any,
     ) -> DataFrame: ...
-    agg = aggregate
+    agg = aggregate  # pyright: ignore[reportUnknownVariableType]
     @overload
     def transform(
         self,

pandas-stubs/core/reshape/api.pyi

@@ -14,11 +14,17 @@ from pandas.core.reshape.merge import (
     merge_ordered as merge_ordered,
 )
 from pandas.core.reshape.pivot import (
-    crosstab as crosstab,
+    crosstab as crosstab,  # pyright: ignore[reportUnknownVariableType]
+)
+from pandas.core.reshape.pivot import (
     pivot as pivot,
-    pivot_table as pivot_table,
+)
+from pandas.core.reshape.pivot import (
+    pivot_table as pivot_table,  # pyright: ignore[reportUnknownVariableType]
+)
+from pandas.core.reshape.tile import (
+    cut as cut,  # pyright: ignore[reportUnknownVariableType]
 )
 from pandas.core.reshape.tile import (
-    cut as cut,
     qcut as qcut,
 )

pandas-stubs/core/reshape/pivot.pyi

@@ -33,7 +33,7 @@ from pandas._typing import (
 
 _PivotAggCallable: TypeAlias = Callable[[Series], ScalarT]
 
-_PivotAggFunc: TypeAlias = (
+_PivotAggFunc: TypeAlias = (  # pyright: ignore[reportUnknownVariableType]
     _PivotAggCallable
     | np.ufunc
     | Literal["mean", "sum", "count", "min", "max", "median", "std", "var"]

pandas-stubs/core/series.pyi

@@ -135,14 +135,15 @@ from pandas._libs.lib import _NoDefaultDoNotUse
 from pandas._libs.missing import NAType
 from pandas._libs.tslibs import BaseOffset
 from pandas._libs.tslibs.nattype import NaTType
+from pandas._typing import (
+    AggFuncTypeSeriesToFrame,  # pyright: ignore[reportUnknownVariableType]
+)
 from pandas._typing import (
     S1,
     S2,
     S2_NSDT,
     T_COMPLEX,
-    AggFuncTypeBase,
     AggFuncTypeDictFrame,
-    AggFuncTypeSeriesToFrame,
     AnyAll,
     AnyArrayLike,
     ArrayLike,
@@ -245,6 +246,7 @@ from pandas._typing import (
     np_ndarray_str,
     np_ndarray_td,
 )
+from pandas._typing import AggFuncTypeBase  # pyright: ignore[reportUnknownVariableType]
 
 from pandas.core.dtypes.base import ExtensionDtype
 from pandas.core.dtypes.dtypes import CategoricalDtype
@@ -1147,7 +1149,7 @@ class Series(IndexOpsMixin[S1], ElementOpsMixin[S1], NDFrame):
         *args: Any,
         **kwargs: Any,
     ) -> Series: ...
-    agg = aggregate
+    agg = aggregate  # pyright: ignore[reportUnknownVariableType]
     @overload
     def transform(
         self,