🐛 return self for edge methods

#8
ehne · Sep 27, 2021 · 886c363 · 886c363
1 parent c402958
commit 886c363
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/pynode_next/edge.py b/pynode_next/edge.py
@@ -49,6 +49,7 @@ def target(self, source=None):
     def set_attribute(self, name, value):
         """Sets an attribute of the edge"""
         self._attrs[name] = value
+        return self
 
     def attribute(self, name):
         """Gets an attribute of the edge"""
@@ -176,6 +177,7 @@ def traverse(self, initial_node=None, color=Color.RED, keep_path=True):
         else:
             # stores the changed color
             self._color = color
+        return self
 
     def __str__(self):
         return f"({self._source}, {self._target})"