FIX: fixed CheckedSession on Python3.14 (see #1147)

Author: gdementen

larray/core/checked.py

@@ -7,6 +7,7 @@
 from larray.core.axis import AxisCollection
 from larray.core.array import Array, full
 from larray.core.session import Session
+from larray.util.misc import get_annotations
 
 
 class NotLoaded:
@@ -132,11 +133,8 @@ def validate_array(value: Any, info: ValidationInfo) -> Array:
     class LArrayModelMetaclass(ModelMetaclass):
         def __new__(mcs, cls_name: str, bases: tuple[type[Any], ...],
                     namespace: dict[str, Any], **kwargs):
-
-            # any type hints defined in the class body will land in
-            # __annotations__ (this is not pydantic-specific) but
-            # __annotations__ is only defined if there are type hints
-            raw_annotations = namespace.get('__annotations__', {})
+            # get user-defined annotations
+            raw_annotations = get_annotations(namespace)
             type_annotations = {
                 key: type(value)
                 for key, value in namespace.items()

larray/util/misc.py

@@ -1081,3 +1081,27 @@ def concatenate_ndarrays(arrays) -> np.ndarray:
 
 def first(iterable, default=None):
     return next(iter(iterable), default)
+
+
+try:
+    # Python 3.14+
+    import annotationlib
+
+    def get_annotations(namespace):
+        # should not happen in Python3.14+ unless
+        # "from __future__ import annotations" is used
+        if "__annotations__" in namespace:
+            return namespace["__annotations__"]
+        elif annotate := annotationlib.get_annotate_from_class_namespace(namespace):
+            return annotationlib.call_annotate_function(
+                annotate, format=annotationlib.Format.FORWARDREF
+            )
+        else:
+            return {}
+except ImportError:
+    # Python <3.14
+    def get_annotations(namespace):
+        # any type hints defined in the class body will land in a
+        # __annotations__ key in its namespace (this is not pydantic-specific)
+        # but __annotations__ is only defined if there are type hints
+        return namespace.get('__annotations__', {})