Refactor model to use property for link number instead of post init

ami-iit · Jan 2, 2025 · c8adbbc · c8adbbc
1 parent 8ddbf04
commit c8adbbc
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/src/adam/model/model.py b/src/adam/model/model.py
@@ -17,9 +17,14 @@ class Model:
     NDoF: int
     actuated_joints: list[str]
 
-    def __post_init__(self):
-        """set the "length of the model as the number of links"""
-        self.N = len(self.links)
+    @property
+    def N(self) -> int:
+        """
+
+        Returns:
+            int: the number of links in the model
+        """
+        return len(self.links)
 
     @staticmethod
     def build(factory: ModelFactory, joints_name_list: list[str] = None) -> "Model":