df1dd9b108f3cd7d5130e7f654ea099021c7a2b3,qanta/guesser/experimental/elasticsearch_instance_of.py,ElasticSearchIndex,search,#ElasticSearchIndex#Any#Any#Any#Any#,82

Before Change

                fields=["wiki_content", "qb_content"]
            )
        results = s.execute()
        return [(r.page, r.meta.score) for r in results]


es_index = ElasticSearchIndex()

After Change

    class_to_i = {label: i for i, label in enumerate(classes, 1)}
    i_to_class = {i: label for label, i in class_to_i.items()}

    for q, p in zip(questions, pages):
        for sent in q:
            x_data.append(sent)
            if p in instance_of_map:
                y_data.append(class_to_i[instance_of_map[p]])
            else:
                y_data.append(class_to_i[NO_MATCH])

Instances