def createNode(self, nodeType, name=None, pos=None): """ overrides Flowchart.createNode to pick sensible names that caffe will accept """ if name is None: name = nodeType.lower() if name in self._nodes: name2 = name n = 1 while name2 in self._nodes: name2 = "%s.%d" % (name, n) n += 1 name = name2 return Flowchart.createNode(self, nodeType, name=name, pos=pos)