make GenericDtype TypeVar bound union of supported types (#960)

Signed-off-by: Niels Bantilan <niels.bantilan@gmail.com> Signed-off-by: Niels Bantilan <niels.bantilan@gmail.com>
unionai-oss · Oct 10, 2022 · 5fd3e55 · 5fd3e55
1 parent 1cc4f86
commit 5fd3e55
Showing 1 changed file with 76 additions and 74 deletions.
diff --git a/pandera/typing/common.py b/pandera/typing/common.py
@@ -2,7 +2,7 @@
 # pylint:disable=abstract-method,disable=too-many-ancestors
 
 import inspect
-from typing import TYPE_CHECKING, Any, Generic, Optional, Type, TypeVar
+from typing import TYPE_CHECKING, Any, Generic, Optional, Type, TypeVar, Union
 
 import pandas as pd
 import typing_inspect
@@ -53,83 +53,85 @@
 if GEOPANDAS_INSTALLED:
     GenericDtype = TypeVar(  # type: ignore
         "GenericDtype",
-        bool,
-        int,
-        str,
-        float,
-        pd.core.dtypes.base.ExtensionDtype,
-        Bool,
-        Date,
-        DateTime,
-        Decimal,
-        Timedelta,
-        Category,
-        Float,
-        Float16,
-        Float32,
-        Float64,
-        Int,
-        Int8,
-        Int16,
-        Int32,
-        Int64,
-        UInt8,
-        UInt16,
-        UInt32,
-        UInt64,
-        INT8,
-        INT16,
-        INT32,
-        INT64,
-        UINT8,
-        UINT16,
-        UINT32,
-        UINT64,
-        Object,
-        String,
-        STRING,
-        Geometry,
-        covariant=True,
+        bound=Union[
+            bool,
+            int,
+            str,
+            float,
+            pd.core.dtypes.base.ExtensionDtype,
+            Bool,
+            Date,
+            DateTime,
+            Decimal,
+            Timedelta,
+            Category,
+            Float,
+            Float16,
+            Float32,
+            Float64,
+            Int,
+            Int8,
+            Int16,
+            Int32,
+            Int64,
+            UInt8,
+            UInt16,
+            UInt32,
+            UInt64,
+            INT8,
+            INT16,
+            INT32,
+            INT64,
+            UINT8,
+            UINT16,
+            UINT32,
+            UINT64,
+            Object,
+            String,
+            STRING,
+            Geometry,
+        ],
     )
 else:
     GenericDtype = TypeVar(  # type: ignore
         "GenericDtype",
-        bool,
-        int,
-        str,
-        float,
-        pd.core.dtypes.base.ExtensionDtype,
-        Bool,
-        Date,
-        DateTime,
-        Decimal,
-        Timedelta,
-        Category,
-        Float,
-        Float16,
-        Float32,
-        Float64,
-        Int,
-        Int8,
-        Int16,
-        Int32,
-        Int64,
-        UInt8,
-        UInt16,
-        UInt32,
-        UInt64,
-        INT8,
-        INT16,
-        INT32,
-        INT64,
-        UINT8,
-        UINT16,
-        UINT32,
-        UINT64,
-        Object,
-        String,
-        STRING,
-        covariant=True,
+        bound=Union[
+            bool,
+            int,
+            str,
+            float,
+            pd.core.dtypes.base.ExtensionDtype,
+            Bool,
+            Date,
+            DateTime,
+            Decimal,
+            Timedelta,
+            Category,
+            Float,
+            Float16,
+            Float32,
+            Float64,
+            Int,
+            Int8,
+            Int16,
+            Int32,
+            Int64,
+            UInt8,
+            UInt16,
+            UInt32,
+            UInt64,
+            INT8,
+            INT16,
+            INT32,
+            INT64,
+            UINT8,
+            UINT16,
+            UINT32,
+            UINT64,
+            Object,
+            String,
+            STRING,
+        ],
     )
 
 Schema = TypeVar("Schema", bound="SchemaModel")  # type: ignore