docstring and fix

compas-dev · Jan 4, 2024 · 01266b7 · 01266b7
1 parent 37f85f0
commit 01266b7
Showing 1 changed file with 6 additions and 10 deletions.
diff --git a/src/compas/datastructures/tree/tree.py b/src/compas/datastructures/tree/tree.py
@@ -13,10 +13,8 @@ class TreeNode(Data):
 
     Parameters
     ----------
-    name : str, optional
-        The name of the tree ndoe.
-    attributes : dict[str, Any], optional
-        User-defined attributes of the datastructure.
+    **kwargs : dict[str, Any], optional
+        User-defined attributes of the tree node.
 
     Attributes
     ----------
@@ -94,7 +92,7 @@ def data(self):
 
     @classmethod
     def from_data(cls, data):
-        node = cls(attributes=data["attributes"])
+        node = cls(**data["attributes"])
         for child in data["children"]:
             node.add(cls.from_data(child))
         return node
@@ -206,10 +204,8 @@ class Tree(Datastructure):
 
     Parameters
     ----------
-    name : str, optional
-        The name of the datastructure.
-    attributes : dict[str, Any], optional
-        User-defined attributes of the datastructure.
+    **kwargs : dict[str, Any], optional
+        User-defined attributes of the tree.
 
     Attributes
     ----------
@@ -264,7 +260,7 @@ def data(self):
 
     @classmethod
     def from_data(cls, data):
-        tree = cls(attributes=data["attributes"])
+        tree = cls(**data["attributes"])
         root = TreeNode.from_data(data["root"])
         tree.add(root)
         return tree