Commit 996ff839 by Eric Junyuan Xie Committed by Tianqi Chen

fix (#87)

parent 390c280b
......@@ -446,8 +446,7 @@ void Symbol::SetAttrs(const std::vector<std::pair<std::string, std::string> >& a
bool Symbol::GetAttr(const std::string& key, std::string* out) const {
Node* node = outputs[0].node.get();
for (const NodeEntry& e : outputs) {
CHECK(node == e.node.get())
<< "Symbol.SetAttrs only works for non-grouped symbol";
if (node != e.node.get()) return false;
}
if (key == "name") {
*out = node->attrs.name;
......
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