♻️ refactor: Use standard typing functions over old typing_inspect

Author: glatterf42

pandera/api/extensions.py

@@ -5,9 +5,7 @@
 from enum import Enum
 from functools import partial, wraps
 from inspect import signature
-from typing import Callable, Optional, Union
-
-import typing_inspect
+from typing import Callable, Optional, Union, get_args, get_origin
 
 from pandera.api.checks import Check
 from pandera.api.hypotheses import Hypothesis
@@ -56,11 +54,11 @@ def register_builtin_check(
     # object to validate is the first argument.
     data_type = [*fn_sig.parameters.values()][0].annotation
 
-    if typing_inspect.get_origin(data_type) is tuple:
-        data_type, *_ = typing_inspect.get_args(data_type)
+    if get_origin(data_type) is tuple:
+        data_type, *_ = get_args(data_type)
 
-    if typing_inspect.get_origin(data_type) is Union:
-        data_types = typing_inspect.get_args(data_type)
+    if get_origin(data_type) is Union:
+        data_types = get_args(data_type)
     else:
         data_types = (data_type,)
 

pandera/api/function_dispatch.py

@@ -1,8 +1,7 @@
 """Multidispatcher implementation."""
 
 from inspect import signature
-from typing import Callable, Union
-import typing_inspect
+from typing import Callable, Union, get_args, get_origin
 
 
 class Dispatcher:
@@ -52,11 +51,11 @@ def get_first_arg_type(fn):
     # object to validate is the first argument.
     data_type = [*fn_sig.parameters.values()][0].annotation
 
-    if typing_inspect.get_origin(data_type) in (tuple, tuple):
-        data_type, *_ = typing_inspect.get_args(data_type)
+    if get_origin(data_type) in (tuple, tuple):
+        data_type, *_ = get_args(data_type)
 
-    if typing_inspect.get_origin(data_type) is Union:
-        data_types = typing_inspect.get_args(data_type)
+    if get_origin(data_type) is Union:
+        data_types = get_args(data_type)
     else:
         data_types = (data_type,)
 

pandera/engines/engine.py

@@ -6,17 +6,9 @@
 import sys
 from abc import ABCMeta
 from dataclasses import dataclass
-from typing import (
-    TYPE_CHECKING,
-    Any,
-    Callable,
-    NamedTuple,
-    Optional,
-    TypeVar,
-)
+from typing import TYPE_CHECKING, Any, Callable, NamedTuple, Optional, TypeVar, get_args, get_origin, get_type_hints
 
 import typing_inspect
-from typing import get_type_hints
 
 from pandera.dtypes import DataType
 
@@ -139,7 +131,7 @@ def _register_from_parametrized_dtype(
         annotations = get_type_hints(func).values()
         dtype = next(iter(annotations))  # get 1st annotation
         # parse typing.Union
-        dtypes = typing_inspect.get_args(dtype) or [dtype]
+        dtypes = get_args(dtype) or [dtype]
 
         def _method(*args, **kwargs):
             return func(pandera_dtype_cls, *args, **kwargs)
@@ -243,7 +235,7 @@ def dtype(cls: "Engine", data_type: Any) -> DataType:
         registry = cls._registry[cls]
 
         # handle python generic types, e.g. typing.Dict[str, str]
-        datatype_origin = typing_inspect.get_origin(data_type)
+        datatype_origin = get_origin(data_type)
         if datatype_origin is not None:
             equivalent_data_type = registry.get_equivalent(datatype_origin)
             return type(equivalent_data_type)(data_type)  # type: ignore

pandera/typing/common.py

@@ -11,6 +11,8 @@
     TypeVar,
     Union,
     _GenericAlias,
+    get_args,
+    get_origin,
 )
 
 import typing_inspect
@@ -217,12 +219,12 @@ def _parse_annotation(self, raw_annotation: type) -> None:
         if self.optional and typing_inspect.is_union_type(raw_annotation):
             # Annotated with Optional or Union[..., NoneType]
             # get_args -> (pandera.typing.Index[str], <class 'NoneType'>)
-            raw_annotation = typing_inspect.get_args(raw_annotation)[0]
+            raw_annotation = get_args(raw_annotation)[0]
             self.raw_annotation = raw_annotation
 
-        self.origin = typing_inspect.get_origin(raw_annotation)
+        self.origin = get_origin(raw_annotation)
         # Replace empty tuple returned from get_args by None
-        args = typing_inspect.get_args(raw_annotation) or None
+        args = get_args(raw_annotation) or None
         self.args = args
         self.arg = args[0] if args else args
 
@@ -236,14 +238,14 @@ def _parse_annotation(self, raw_annotation: type) -> None:
                 metadata = None
 
         elif metadata := getattr(self.arg, "__metadata__", None):
-            self.arg = typing_inspect.get_args(self.arg)[0]
+            self.arg = get_args(self.arg)[0]
 
         self.metadata = metadata
 
-        self.literal = typing_inspect.get_origin(self.arg) is typing.Literal
+        self.literal = get_origin(self.arg) is typing.Literal
 
         if self.literal:
-            self.arg = typing_inspect.get_args(self.arg)[0]
+            self.arg = get_args(self.arg)[0]
         elif self.origin is None and self.metadata is None:
             if isinstance(raw_annotation, type) and issubclass(
                 raw_annotation, SeriesBase