Avoid throwing exception on Union types (#1378)

Signed-off-by: Matthew Painter <[email protected]>
unionai-oss · Oct 20, 2023 · fb680ac · fb680ac
1 parent fc86c4b
commit fb680ac
Showing 1 changed file with 1 addition and 5 deletions.
diff --git a/pandera/mypy.py b/pandera/mypy.py
@@ -136,7 +136,7 @@ class Schema(BaseSchema):
             if not hasattr(def_, "type") or def_.type is None:
                 continue
             type_ = def_.type
-            if get_typename(def_.type) in FIELD_GENERICS_FULLNAMES:
+            if str(def_.type) in FIELD_GENERICS_FULLNAMES:
                 type_.args = ()  # erase generic type arg
 
 
@@ -154,7 +154,3 @@ def get_fullname(x: Union[FuncBase, SymbolNode]) -> str:
     if callable(fn):  # pragma: no cover
         return fn()
     return fn
-
-
-def get_typename(x) -> str:
-    return f"{x.type.module_name}.{x.type.name}"