fix compat for types.UnionType

Author: voidZXL

utype/parser/rule.py

@@ -1,6 +1,4 @@
-import inspect
 import re
-import types
 import typing
 import warnings
 from collections import deque
@@ -12,7 +10,7 @@
 
 from ..utils import exceptions as exc
 from ..utils.compat import (ForwardRef, Literal, evaluate_forward_ref,
-                            get_args, get_origin)
+                            get_args, get_origin, UnionType)
 from ..utils.datastructures import unprovided
 from ..utils.functional import multi, pop
 from ..utils.transform import TypeTransformer
@@ -1308,7 +1306,7 @@ def annotate(
         #     return type_
 
         name = cls.__name__
-        if type_ == Union or type_ == types.UnionType:
+        if type_ == Union or type_ == UnionType:
             # in Python >= 3.10, native logic operator like int | str will be a UnionType
             type_ = LogicalType.any_of(*args)
             # clear the args for union

utype/utils/compat.py

@@ -11,6 +11,11 @@
 except ImportError:
     from typing_extensions import Literal
 
+try:
+    from types import UnionType
+except ImportError:
+    UnionType = Union
+
 try:
     from typing import Final
 except ImportError:
@@ -27,6 +32,7 @@
     "get_args",
     'Literal',
     'Final',
+    'UnionType',
     "ForwardRef",
     "Annotated",
     "is_final",