rolled back some type, converted back to Type, in pandas & polars engine

Author: g-as

pandera/engines/pandas_engine.py

@@ -21,6 +21,7 @@
     Literal,
     NamedTuple,
     Optional,
+    Type,
     Union,
     cast,
 )
@@ -1216,11 +1217,11 @@ def from_parametrized_dtype(cls, pd_dtype: pd.IntervalDtype):
 class PydanticModel(DataType):
     """A pydantic model datatype applying to rows in a dataframe."""
 
-    type: type[BaseModel] = dataclasses.field(default=None, init=False)  # type: ignore[assignment]
+    type: Type[BaseModel] = dataclasses.field(default=None, init=False)  # type: ignore[assignment]
     auto_coerce = True
 
     # pylint:disable=super-init-not-called
-    def __init__(self, model: type[BaseModel]) -> None:
+    def __init__(self, model: Type[BaseModel]) -> None:
         object.__setattr__(self, "type", model)
 
     def coerce(self, data_container: PandasObject) -> PandasObject:
@@ -1281,7 +1282,7 @@ class PythonGenericType(DataType):
     type: Any = dataclasses.field(default=None, init=False)  # type: ignore
     generic_type: Any = dataclasses.field(default=None, init=False)
     special_type: Any = dataclasses.field(default=None, init=False)
-    coercion_model: type[BaseModel] = dataclasses.field(  # type: ignore
+    coercion_model: Type[BaseModel] = dataclasses.field(  # type: ignore
         default=None, init=False
     )
     _pandas_type = object

pandera/engines/polars_engine.py

@@ -5,12 +5,7 @@
 import decimal
 import inspect
 import warnings
-from typing import (
-    Any,
-    Literal,
-    Optional,
-    Union,
-)
+from typing import Any, Literal, Optional, Union, Type
 from collections.abc import Iterable, Mapping, Sequence
 
 import polars as pl
@@ -229,8 +224,7 @@ def dtype(cls, data_type: Any) -> dtypes.DataType:
                 pl_dtype = convert_py_dtype_to_polars_dtype(data_type)
             except ValueError:
                 raise TypeError(
-                    f"data type '{data_type}' not understood by "
-                    f"{cls.__name__}."
+                    f"data type '{data_type}' not understood by {cls.__name__}."
                 ) from None
 
             try:
@@ -455,7 +449,7 @@ class Date(DataType, dtypes.Date):
 class DateTime(DataType, dtypes.DateTime):
     """Polars datetime data type."""
 
-    type: type[pl.Datetime] = pl.Datetime
+    type: Type[pl.Datetime] = pl.Datetime
     time_zone_agnostic: bool = False
 
     def __init__(  # pylint:disable=super-init-not-called
@@ -464,7 +458,6 @@ def __init__(  # pylint:disable=super-init-not-called
         time_zone: Optional[str] = None,
         time_unit: Optional[str] = None,
     ) -> None:
-
         _kwargs = {}
         if time_unit is not None:
             # avoid deprecated warning when initializing pl.Datetime:
@@ -566,7 +559,6 @@ def __init__(  # pylint:disable=super-init-not-called
         *,
         width: Optional[int] = None,
     ) -> None:
-
         kwargs: dict[str, Union[int, tuple[int, ...]]] = {}
         if width is not None:
             kwargs["shape"] = width
@@ -713,8 +705,7 @@ def check(
             return False
 
         return (
-            self.type == pandera_dtype.type
-            and (self.type.categories == pandera_dtype.categories).all()  # type: ignore
+            self.type == pandera_dtype.type and (self.type.categories == pandera_dtype.categories).all()  # type: ignore
         )