ac730aec61523031016afc8922c04bfc7a7ac42b,fairseq/sequence_generator.py,SequenceGenerator,_generate,#SequenceGenerator#Any#Any#Any#Any#,168

Before Change

            BCList.sort()
            BCList.reverse()
            finalized[sent] = torch.jit.annotate(
                List[Dict[str, Tensor]], [x.elem for x in BCList]
            )

        return finalized

After Change

        // sort by score descending
        for sent in range(len(finalized)):
            scores = torch.tensor([float(elem["score"].item()) for elem in finalized[sent]])
            _, sorted_scores_indices = torch.sort(scores, descending=True)
            finalized[sent] = [finalized[sent][ssi] for ssi in sorted_scores_indices]
            finalized[sent] = torch.jit.annotate(List[Dict[str, Tensor]], finalized[sent])
        return finalized

Instances