diff --git a/include/aidge/graph/GraphView.hpp b/include/aidge/graph/GraphView.hpp index 682634015376bf309a015046decfa40a36e2b177..8b58d13bc4f249d21199ed1a867baef80b534aac 100644 --- a/include/aidge/graph/GraphView.hpp +++ b/include/aidge/graph/GraphView.hpp @@ -400,10 +400,15 @@ public: addChild(toOtherNode, mNodeRegistry.at(fromOutNodeName), fromTensor, toTensor); } - inline void updateNodeName(const std::string& oldName, const std::string& newName){ - AIDGE_ASSERT(mNodeRegistry.find(oldName) != mNodeRegistry.end(), "No node named {} in graph {}, the graph may be corrupted !", oldName, name()); - mNodeRegistry[newName] = mNodeRegistry[oldName]; - mNodeRegistry.erase(oldName); + inline void updateNodeName(NodePtr nodeToRename, const std::string& newName){ + const std::string& oldName = nodeToRename->name(); + if (nodeToRename->name() != ""){ // Case node already had a name + AIDGE_ASSERT(mNodeRegistry.find(oldName) != mNodeRegistry.end(), "No node named {} in graph {}, the graph may be corrupted !", oldName, name()); + mNodeRegistry[newName] = mNodeRegistry[oldName]; + mNodeRegistry.erase(oldName); + }else{ // Case node did not had a name + mNodeRegistry[newName] = nodeToRename; + } } /** diff --git a/src/graph/Node.cpp b/src/graph/Node.cpp index 50b8be13c6faac62a8b2aecf1e767e0b83024d3c..7fe155b5a2b9b42a1504dbb592b2326d13b99c1e 100644 --- a/src/graph/Node.cpp +++ b/src/graph/Node.cpp @@ -64,7 +64,7 @@ Aidge::Connector Aidge::Node::operator()(const std::vector<Connector>& ctors) { /////////////////////////////////////////////////////// void Aidge::Node::setName(const std::string& name) { - for (auto graphView : views()) graphView->updateNodeName(mName, name); + for (auto graphView : views()) graphView->updateNodeName(shared_from_this(), name); mName = name; }