Commit 2bfa0964 authored by Adam Rogowiec's avatar Adam Rogowiec Committed by Scott Cyphers

Use get_friendly_name function to get Node name. (#1017)

parent f56e4766
......@@ -59,7 +59,7 @@ void regclass_pyngraph_Node(py::module m)
node.def("__repr__", [](const ngraph::Node& self) {
std::string class_name = py::cast(self).get_type().attr("__name__").cast<std::string>();
std::string shape = py::cast(self.get_shape()).attr("__str__")().cast<std::string>();
return "<" + class_name + ": '" + self.get_name() + "' (" + shape + ")>";
return "<" + class_name + ": '" + self.get_friendly_name() + "' (" + shape + ")>";
});
node.def("get_output_size", &ngraph::Node::get_output_size);
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment