fix bug in network

Author: tomvanmele

src/compas/datastructures/graph/graph.py

@@ -366,7 +366,7 @@ def index_uv(self):
     # builders
     # --------------------------------------------------------------------------
 
-    def add_node(self, key, attr_dict=None, **kwattr):
+    def add_node(self, key=None, attr_dict=None, **kwattr):
         """Add a node and specify its attributes (optional).
 
         Parameters
@@ -381,7 +381,7 @@ def add_node(self, key, attr_dict=None, **kwattr):
 
         Returns
         -------
-        str
+        hashable
             The key of the node.
 
         Notes
@@ -396,6 +396,14 @@ def add_node(self, key, attr_dict=None, **kwattr):
         --------
         >>>
         """
+        if key is None:
+            key = self._max_node = self._max_node + 1
+        try:
+            if key > self._max_node:
+                self._max_node = key
+        except (ValueError, TypeError):
+            pass
+
         if key not in self.node:
             self.node[key] = {}
             self.edge[key] = {}

src/compas/datastructures/network/network.py

@@ -327,16 +327,6 @@ def gkey_key(self, precision=None):
     # builders
     # --------------------------------------------------------------------------
 
-    def add_node(self, key=None, attr_dict=None, **kwattr):
-        if key is None:
-            key = self._max_int_key = self._max_int_key + 1
-        try:
-            if key > self._max_int_key:
-                self._max_int_key = key
-        except (ValueError, TypeError):
-            pass
-        return super(Network, self).add_node(key, attr_dict=attr_dict, **kwattr)
-
     # --------------------------------------------------------------------------
     # modifiers
     # --------------------------------------------------------------------------