FIX: fixed calling methods on CheckedSession subclasses

(thanks to Alix for finding the solution to this)

Author: gdementen

larray/core/checked.py

@@ -116,9 +116,14 @@ def __new__(mcs, cls_name: str, bases: tuple[type[Any], ...], namespace: dict[st
             raw_annotations = namespace.get('__annotations__', {})
 
             # tries to infer types for variables without type hints
-            keys_to_infer_type = [key for key in namespace.keys() if key not in raw_annotations]
-            keys_to_infer_type = [key for key in keys_to_infer_type if is_valid_field_name(key)]
-            keys_to_infer_type = [key for key in keys_to_infer_type if key not in {'model_config', 'dict'}]
+            keys_to_infer_type = [key for key in namespace.keys()
+                                  if key not in raw_annotations]
+            keys_to_infer_type = [key for key in keys_to_infer_type
+                                  if is_valid_field_name(key)]
+            keys_to_infer_type = [key for key in keys_to_infer_type
+                                  if key not in {'model_config', 'dict', 'build'}]
+            keys_to_infer_type = [key for key in keys_to_infer_type
+                                  if not callable(namespace[key])]
             for key in keys_to_infer_type:
                 value = namespace[key]
                 raw_annotations[key] = type(value)

larray/tests/test_checked_session.py

@@ -715,5 +715,24 @@ def test_neg_cs(checkedsession):
     assert_array_nan_equal(neg_cs.h, -h)
 
 
+def test_checked_class_with_methods():
+    a = Axis('a=a0,a1')
+
+    class CheckedSessionWithMethods(CheckedSession):
+        arr: CheckedArray(a)
+
+        # Define a method which already exists in Session/CheckedSession
+        def save(self, path=None, **kwargs):
+            super().save(path, **kwargs)
+
+        def new_method(self):
+            return True
+
+    array = ndtest(a)
+
+    cs = CheckedSessionWithMethods(arr=array)
+    assert cs.new_method()
+
+
 if __name__ == "__main__":
     pytest.main()