def create_node_with_nodes(nodes, name='node'): node = hsm_pb2.NodeProto() node.name = name for child_node in nodes: new_child_node = node.children.add() new_child_node.MergeFrom(child_node) return node
def create_node_with_words(words, name='node'): node = hsm_pb2.NodeProto() node.name = name for word in words: node.word_ids.append(word) return node
def create_node_with_words(words): node = hsm_pb2.NodeProto() for word in words: node.word_ids.append(word) return node