fix: error type hints in Python3.12 (#1147)

954-Ivory · 954-Ivory · commit 9391424c192c · 2023-12-22T17:38:04.000+08:00
Signed-off-by: 954 &lt;510485871@qq.com&gt;
diff --git a/docarray/array/any_array.py b/docarray/array/any_array.py
@@ -1,3 +1,4 @@
+import sys
 import random
 from abc import abstractmethod
 from typing import (
@@ -29,6 +30,9 @@
     from docarray.proto import DocListProto, NodeProto
     from docarray.typing.tensor.abstract_tensor import AbstractTensor
 
+if sys.version_info < (3, 12):
+    from types import GenericAlias
+
 T = TypeVar('T', bound='AnyDocArray')
 T_doc = TypeVar('T_doc', bound=BaseDocWithoutId)
 IndexIterType = Union[slice, Iterable[int], Iterable[bool], None]
@@ -51,8 +55,12 @@ def __repr__(self):
     @classmethod
     def __class_getitem__(cls, item: Union[Type[BaseDocWithoutId], TypeVar, str]):
         if not isinstance(item, type):
-            return Generic.__class_getitem__.__func__(cls, item)  # type: ignore
-            # this do nothing that checking that item is valid type var or str
+            if sys.version_info < (3, 12):
+                return Generic.__class_getitem__.__func__(cls, item)  # type: ignore
+                # this do nothing that checking that item is valid type var or str
+                # Keep the approach in #1147 to be compatible with lower versions of Python.
+            else:
+                return GenericAlias(cls, item)  # type: ignore
         if not safe_issubclass(item, BaseDocWithoutId):
             raise ValueError(
                 f'{cls.__name__}[item] item should be a Document not a {item} '
