TYP: enable reportGeneralTypeIssues but add many exceptions

pandas-dev · Mar 1, 2022 · 1409004 · 1409004
1 parent 53b3dd5
commit 1409004
Show file tree

Hide file tree

Showing 153 changed files with 153 additions and 4 deletions.
diff --git a/.github/workflows/code-checks.yml b/.github/workflows/code-checks.yml
@@ -74,7 +74,7 @@ jobs:
 
     - name: Install pyright
       # note: keep version in sync with .pre-commit-config.yaml
-      run: npm install -g pyright@1.1.212
+      run: npm install -g pyright@1.1.225
 
     - name: Build Pandas
       id: build

diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
@@ -85,7 +85,7 @@ repos:
         types: [python]
         stages: [manual]
         # note: keep version in sync with .github/workflows/code-checks.yml
-        additional_dependencies: ['pyright@1.1.212']
+        additional_dependencies: ['pyright@1.1.225']
 -   repo: local
     hooks:
     -   id: flake8-rst

diff --git a/pandas/_config/config.py b/pandas/_config/config.py
@@ -1,3 +1,4 @@
+# pyright: reportGeneralTypeIssues = false
 """
 The config module holds package-wide configurables and provides
 a uniform API for working with them.

diff --git a/pandas/_config/localization.py b/pandas/_config/localization.py
@@ -1,3 +1,4 @@
+# pyright: reportGeneralTypeIssues = false
 """
 Helpers for configuring locale settings.
 

diff --git a/pandas/_testing/__init__.py b/pandas/_testing/__init__.py
@@ -1,3 +1,4 @@
+# pyright: reportGeneralTypeIssues = false
 from __future__ import annotations
 
 import collections

diff --git a/pandas/_testing/_io.py b/pandas/_testing/_io.py
@@ -1,3 +1,4 @@
+# pyright: reportGeneralTypeIssues = false
 from __future__ import annotations
 
 import bz2

diff --git a/pandas/_testing/asserters.py b/pandas/_testing/asserters.py
@@ -1,3 +1,4 @@
+# pyright: reportGeneralTypeIssues = false
 from __future__ import annotations
 
 from typing import cast

diff --git a/pandas/compat/pickle_compat.py b/pandas/compat/pickle_compat.py
@@ -1,3 +1,4 @@
+# pyright: reportGeneralTypeIssues = false
 """
 Support pre-0.12 series pickle compatibility.
 """

diff --git a/pandas/conftest.py b/pandas/conftest.py
@@ -1,3 +1,4 @@
+# pyright: reportGeneralTypeIssues = false
 """
 This file is very long and growing, but it was decided to not split it yet, as
 it's still manageable (2020-03-17, ~1.1k LoC). See gh-31989

diff --git a/pandas/core/algorithms.py b/pandas/core/algorithms.py
@@ -1,3 +1,4 @@
+# pyright: reportGeneralTypeIssues = false
 """
 Generic data algorithms. This module is experimental at the moment and not
 intended for public consumption

diff --git a/pandas/core/apply.py b/pandas/core/apply.py
@@ -1,3 +1,4 @@
+# pyright: reportGeneralTypeIssues = false
 from __future__ import annotations
 
 import abc

diff --git a/pandas/core/array_algos/masked_reductions.py b/pandas/core/array_algos/masked_reductions.py
@@ -1,3 +1,4 @@
+# pyright: reportGeneralTypeIssues = false
 """
 masked_reductions.py is for reduction algorithms using a mask-based approach
 for missing values.

diff --git a/pandas/core/array_algos/quantile.py b/pandas/core/array_algos/quantile.py
@@ -1,3 +1,4 @@
+# pyright: reportGeneralTypeIssues = false
 from __future__ import annotations
 
 import numpy as np

diff --git a/pandas/core/array_algos/replace.py b/pandas/core/array_algos/replace.py
@@ -1,3 +1,4 @@
+# pyright: reportGeneralTypeIssues = false
 """
 Methods used by Block.replace and related methods.
 """

diff --git a/pandas/core/array_algos/take.py b/pandas/core/array_algos/take.py
@@ -1,3 +1,4 @@
+# pyright: reportGeneralTypeIssues = false
 from __future__ import annotations
 
 import functools

diff --git a/pandas/core/arrays/_mixins.py b/pandas/core/arrays/_mixins.py
@@ -1,3 +1,4 @@
+# pyright: reportGeneralTypeIssues = false
 from __future__ import annotations
 
 from functools import wraps

diff --git a/pandas/core/arrays/_ranges.py b/pandas/core/arrays/_ranges.py
@@ -1,3 +1,4 @@
+# pyright: reportGeneralTypeIssues = false
 """
 Helper functions to generate range-like data for DatetimeArray
 (and possibly TimedeltaArray/PeriodArray)

diff --git a/pandas/core/arrays/base.py b/pandas/core/arrays/base.py
@@ -1,3 +1,4 @@
+# pyright: reportGeneralTypeIssues = false
 """
 An interface for extending pandas with custom arrays.
 

diff --git a/pandas/core/arrays/boolean.py b/pandas/core/arrays/boolean.py
@@ -1,3 +1,4 @@
+# pyright: reportGeneralTypeIssues = false
 from __future__ import annotations
 
 import numbers

diff --git a/pandas/core/arrays/categorical.py b/pandas/core/arrays/categorical.py
@@ -1,3 +1,4 @@
+# pyright: reportGeneralTypeIssues = false
 from __future__ import annotations
 
 from csv import QUOTE_NONNUMERIC

diff --git a/pandas/core/arrays/datetimelike.py b/pandas/core/arrays/datetimelike.py
@@ -1,3 +1,4 @@
+# pyright: reportGeneralTypeIssues = false
 from __future__ import annotations
 
 from datetime import (

diff --git a/pandas/core/arrays/datetimes.py b/pandas/core/arrays/datetimes.py
@@ -1,3 +1,4 @@
+# pyright: reportGeneralTypeIssues = false
 from __future__ import annotations
 
 from datetime import (

diff --git a/pandas/core/arrays/interval.py b/pandas/core/arrays/interval.py
@@ -1,3 +1,4 @@
+# pyright: reportGeneralTypeIssues = false
 from __future__ import annotations
 
 import operator

diff --git a/pandas/core/arrays/masked.py b/pandas/core/arrays/masked.py
@@ -1,3 +1,4 @@
+# pyright: reportGeneralTypeIssues = false
 from __future__ import annotations
 
 from typing import (

diff --git a/pandas/core/arrays/numeric.py b/pandas/core/arrays/numeric.py
@@ -1,3 +1,4 @@
+# pyright: reportGeneralTypeIssues = false
 from __future__ import annotations
 
 import numbers

diff --git a/pandas/core/arrays/numpy_.py b/pandas/core/arrays/numpy_.py
@@ -1,3 +1,4 @@
+# pyright: reportGeneralTypeIssues = false
 from __future__ import annotations
 
 import numpy as np

diff --git a/pandas/core/arrays/period.py b/pandas/core/arrays/period.py
@@ -1,3 +1,4 @@
+# pyright: reportGeneralTypeIssues = false
 from __future__ import annotations
 
 from datetime import timedelta

diff --git a/pandas/core/arrays/sparse/accessor.py b/pandas/core/arrays/sparse/accessor.py
@@ -1,3 +1,4 @@
+# pyright: reportGeneralTypeIssues = false
 """Sparse accessor"""
 
 import numpy as np

diff --git a/pandas/core/arrays/sparse/array.py b/pandas/core/arrays/sparse/array.py
@@ -1,3 +1,4 @@
+# pyright: reportGeneralTypeIssues = false
 """
 SparseArray data structure
 """

diff --git a/pandas/core/arrays/sparse/dtype.py b/pandas/core/arrays/sparse/dtype.py
@@ -1,3 +1,4 @@
+# pyright: reportGeneralTypeIssues = false
 """Sparse Dtype"""
 from __future__ import annotations
 

diff --git a/pandas/core/arrays/sparse/scipy_sparse.py b/pandas/core/arrays/sparse/scipy_sparse.py
@@ -1,3 +1,4 @@
+# pyright: reportGeneralTypeIssues = false
 """
 Interaction with scipy.sparse matrices.
 

diff --git a/pandas/core/arrays/string_.py b/pandas/core/arrays/string_.py
@@ -1,3 +1,4 @@
+# pyright: reportGeneralTypeIssues = false
 from __future__ import annotations
 
 from typing import (

diff --git a/pandas/core/arrays/string_arrow.py b/pandas/core/arrays/string_arrow.py
@@ -1,3 +1,4 @@
+# pyright: reportGeneralTypeIssues = false
 from __future__ import annotations
 
 from collections.abc import Callable  # noqa: PDF001

diff --git a/pandas/core/arrays/timedeltas.py b/pandas/core/arrays/timedeltas.py
@@ -1,3 +1,4 @@
+# pyright: reportGeneralTypeIssues = false
 from __future__ import annotations
 
 from datetime import timedelta

diff --git a/pandas/core/base.py b/pandas/core/base.py
@@ -1,3 +1,4 @@
+# pyright: reportGeneralTypeIssues = false
 """
 Base and utility classes for pandas objects.
 """

diff --git a/pandas/core/common.py b/pandas/core/common.py
@@ -1,3 +1,4 @@
+# pyright: reportGeneralTypeIssues = false
 """
 Misc tools for implementing data structures
 

diff --git a/pandas/core/computation/align.py b/pandas/core/computation/align.py
@@ -1,3 +1,4 @@
+# pyright: reportGeneralTypeIssues = false
 """
 Core eval alignment algorithms.
 """

diff --git a/pandas/core/computation/expr.py b/pandas/core/computation/expr.py
@@ -1,3 +1,4 @@
+# pyright: reportGeneralTypeIssues = false
 """
 :func:`~pandas.eval` parsers.
 """

diff --git a/pandas/core/computation/ops.py b/pandas/core/computation/ops.py
@@ -1,3 +1,4 @@
+# pyright: reportGeneralTypeIssues = false
 """
 Operator classes for eval.
 """

diff --git a/pandas/core/computation/pytables.py b/pandas/core/computation/pytables.py
@@ -1,3 +1,4 @@
+# pyright: reportGeneralTypeIssues = false
 """ manage PyTables query interface via Expressions """
 from __future__ import annotations
 

diff --git a/pandas/core/construction.py b/pandas/core/construction.py
@@ -1,3 +1,4 @@
+# pyright: reportGeneralTypeIssues = false
 """
 Constructor functions intended to be shared by pd.array, Series.__init__,
 and Index.__new__.

diff --git a/pandas/core/describe.py b/pandas/core/describe.py
@@ -1,3 +1,4 @@
+# pyright: reportGeneralTypeIssues = false
 """
 Module responsible for execution of NDFrame.describe() method.
 

diff --git a/pandas/core/dtypes/astype.py b/pandas/core/dtypes/astype.py
@@ -1,3 +1,4 @@
+# pyright: reportGeneralTypeIssues = false
 """
 Functions for implementing 'astype' methods according to pandas conventions,
 particularly ones that differ from numpy.

diff --git a/pandas/core/dtypes/base.py b/pandas/core/dtypes/base.py
@@ -1,3 +1,4 @@
+# pyright: reportGeneralTypeIssues = false
 """
 Extend pandas with custom array types.
 """

diff --git a/pandas/core/dtypes/cast.py b/pandas/core/dtypes/cast.py
@@ -1,3 +1,4 @@
+# pyright: reportGeneralTypeIssues = false
 """
 Routines for casting.
 """

diff --git a/pandas/core/dtypes/common.py b/pandas/core/dtypes/common.py
@@ -1,3 +1,4 @@
+# pyright: reportGeneralTypeIssues = false
 """
 Common type operations.
 """

diff --git a/pandas/core/dtypes/concat.py b/pandas/core/dtypes/concat.py
@@ -1,3 +1,4 @@
+# pyright: reportGeneralTypeIssues = false
 """
 Utility functions related to concat.
 """

diff --git a/pandas/core/dtypes/dtypes.py b/pandas/core/dtypes/dtypes.py
@@ -1,3 +1,4 @@
+# pyright: reportGeneralTypeIssues = false
 """
 Define extension dtypes.
 """

diff --git a/pandas/core/frame.py b/pandas/core/frame.py
@@ -1,3 +1,4 @@
+# pyright: reportGeneralTypeIssues = false
 """
 DataFrame
 ---------

diff --git a/pandas/core/generic.py b/pandas/core/generic.py
@@ -1,3 +1,4 @@
+# pyright: reportGeneralTypeIssues = false
 # pyright: reportPropertyTypeMismatch=false
 from __future__ import annotations
 

diff --git a/pandas/core/groupby/categorical.py b/pandas/core/groupby/categorical.py
@@ -1,3 +1,4 @@
+# pyright: reportGeneralTypeIssues = false
 from __future__ import annotations
 
 import numpy as np

diff --git a/pandas/core/groupby/generic.py b/pandas/core/groupby/generic.py
@@ -1,3 +1,4 @@
+# pyright: reportGeneralTypeIssues = false
 """
 Define the SeriesGroupBy and DataFrameGroupBy
 classes that hold the groupby interfaces (and some implementations).

diff --git a/pandas/core/groupby/groupby.py b/pandas/core/groupby/groupby.py
@@ -1,3 +1,4 @@
+# pyright: reportGeneralTypeIssues = false
 """
 Provide the groupby split-apply-combine paradigm. Define the GroupBy
 class providing the base-class of operations.

diff --git a/pandas/core/groupby/grouper.py b/pandas/core/groupby/grouper.py
@@ -1,3 +1,4 @@
+# pyright: reportGeneralTypeIssues = false
 """
 Provide user facing operators for doing the split part of the
 split-apply-combine paradigm.

diff --git a/pandas/core/groupby/ops.py b/pandas/core/groupby/ops.py
@@ -1,3 +1,4 @@
+# pyright: reportGeneralTypeIssues = false
 """
 Provide classes to perform the groupby aggregate operations.
 

diff --git a/pandas/core/indexers/utils.py b/pandas/core/indexers/utils.py
@@ -1,3 +1,4 @@
+# pyright: reportGeneralTypeIssues = false
 """
 Low-dependency indexing utilities.
 """

diff --git a/pandas/core/indexes/accessors.py b/pandas/core/indexes/accessors.py
@@ -1,3 +1,4 @@
+# pyright: reportGeneralTypeIssues = false
 """
 datetimelike delegation
 """

diff --git a/pandas/core/indexes/api.py b/pandas/core/indexes/api.py
@@ -1,3 +1,4 @@
+# pyright: reportGeneralTypeIssues = false
 from __future__ import annotations
 
 import textwrap

diff --git a/pandas/core/indexes/base.py b/pandas/core/indexes/base.py
@@ -1,3 +1,4 @@
+# pyright: reportGeneralTypeIssues = false
 from __future__ import annotations
 
 from datetime import datetime

diff --git a/pandas/core/indexes/category.py b/pandas/core/indexes/category.py
@@ -1,3 +1,4 @@
+# pyright: reportGeneralTypeIssues = false
 from __future__ import annotations
 
 from typing import (

diff --git a/pandas/core/indexes/datetimelike.py b/pandas/core/indexes/datetimelike.py
@@ -1,3 +1,4 @@
+# pyright: reportGeneralTypeIssues = false
 """
 Base and utility classes for tseries type pandas objects.
 """

diff --git a/pandas/core/indexes/datetimes.py b/pandas/core/indexes/datetimes.py
@@ -1,3 +1,4 @@
+# pyright: reportGeneralTypeIssues = false
 from __future__ import annotations
 
 from datetime import (

diff --git a/pandas/core/indexes/extension.py b/pandas/core/indexes/extension.py
@@ -1,3 +1,4 @@
+# pyright: reportGeneralTypeIssues = false
 """
 Shared methods for Index subclasses backed by ExtensionArray.
 """

diff --git a/pandas/core/indexes/frozen.py b/pandas/core/indexes/frozen.py
@@ -1,3 +1,4 @@
+# pyright: reportGeneralTypeIssues = false
 """
 frozen (immutable) data structures to support MultiIndexing
 

diff --git a/pandas/core/indexes/interval.py b/pandas/core/indexes/interval.py
@@ -1,3 +1,4 @@
+# pyright: reportGeneralTypeIssues = false
 """ define the IntervalIndex """
 from __future__ import annotations
 

diff --git a/pandas/core/indexes/multi.py b/pandas/core/indexes/multi.py
@@ -1,3 +1,4 @@
+# pyright: reportGeneralTypeIssues = false
 from __future__ import annotations
 
 from functools import wraps

diff --git a/pandas/core/indexes/numeric.py b/pandas/core/indexes/numeric.py
@@ -1,3 +1,4 @@
+# pyright: reportGeneralTypeIssues = false
 from __future__ import annotations
 
 from typing import (

diff --git a/pandas/core/indexes/period.py b/pandas/core/indexes/period.py
@@ -1,3 +1,4 @@
+# pyright: reportGeneralTypeIssues = false
 from __future__ import annotations
 
 from datetime import (

diff --git a/pandas/core/indexes/range.py b/pandas/core/indexes/range.py
@@ -1,3 +1,4 @@
+# pyright: reportGeneralTypeIssues = false
 from __future__ import annotations
 
 from datetime import timedelta

diff --git a/pandas/core/indexes/timedeltas.py b/pandas/core/indexes/timedeltas.py
@@ -1,3 +1,4 @@
+# pyright: reportGeneralTypeIssues = false
 """ implement the TimedeltaIndex """
 from __future__ import annotations
 

diff --git a/pandas/core/indexing.py b/pandas/core/indexing.py
@@ -1,3 +1,4 @@
+# pyright: reportGeneralTypeIssues = false
 from __future__ import annotations
 
 from contextlib import suppress

diff --git a/pandas/core/internals/api.py b/pandas/core/internals/api.py
@@ -1,3 +1,4 @@
+# pyright: reportGeneralTypeIssues = false
 """
 This is a pseudo-public API for downstream libraries.  We ask that downstream
 authors

diff --git a/pandas/core/internals/array_manager.py b/pandas/core/internals/array_manager.py
@@ -1,3 +1,4 @@
+# pyright: reportGeneralTypeIssues = false
 """
 Experimental manager based on storing a collection of 1D arrays
 """

diff --git a/pandas/core/internals/base.py b/pandas/core/internals/base.py
@@ -1,3 +1,4 @@
+# pyright: reportGeneralTypeIssues = false
 """
 Base class for the internal managers. Both BlockManager and ArrayManager
 inherit from this class.

diff --git a/pandas/core/internals/blocks.py b/pandas/core/internals/blocks.py
@@ -1,3 +1,4 @@
+# pyright: reportGeneralTypeIssues = false
 from __future__ import annotations
 
 from functools import wraps

diff --git a/pandas/core/internals/concat.py b/pandas/core/internals/concat.py
@@ -1,3 +1,4 @@
+# pyright: reportGeneralTypeIssues = false
 from __future__ import annotations
 
 import itertools

diff --git a/pandas/core/internals/construction.py b/pandas/core/internals/construction.py
@@ -1,3 +1,4 @@
+# pyright: reportGeneralTypeIssues = false
 """
 Functions for preparing various inputs passed to the DataFrame or Series
 constructors before passing them to a BlockManager.

diff --git a/pandas/core/internals/managers.py b/pandas/core/internals/managers.py
@@ -1,3 +1,4 @@
+# pyright: reportGeneralTypeIssues = false
 from __future__ import annotations
 
 import itertools