def __call__(self, query_texts, return_score=False):
    if self.use_ed:
        (preds, score) = self.ed(query_texts, self.dictionary)
    if self.use_trie:
        (preds, score) = self.trie(query_texts, self.dictionary)
    if return_score:
        return (preds, score)
    else:
        return preds
def __call__(self, query_texts, return_score=False):
    if self.use_ed:
        (preds, score) = self.ed(query_texts, self.dictionary)
    if self.use_trie:
        (preds, score) = self.trie(query_texts, self.dictionary)
    if return_score:
        return (preds, score)
    else:
        return preds