d7bcbade517750cc4fdb7642415e4cf05ae584e8,tutorial/utils.py,,collect_pubtator_annotations,#,8

Before Change

    Given a list of ddlite Documents with PubTator/CDR annotations,
    extract a dictionary of annotations by type.
    
    annotations = defaultdict(list)
    for a in doc.attribs["root"].xpath(".//annotation"):

        // Relation annotations
        if len(a.xpath("./infon[@key="relation"]")) > 0:
            type = a.xpath("./infon[@key="relation"]/text()")[0]
            types = a.xpath("./infon[@key != "relation"]/@key")
            mesh_ids = a.xpath("./infon[@key != "relation"]/text()")
            annotations[type].append(PubtatorRelation(types=types, mesh_ids=mesh_ids))

        // Mention annotations
        else:
            txt = a.xpath("./text/text()")[0]

After Change

                    si = i - 1
                    break
            //offset -= sent_offsets[si]
            ngrams.append(Ngram(offset, offset + length - 1, sents[si], metadata={
                "mesh_id" : mesh, "type" : type}))
    return ngrams

Instances