ceda7ee4c7659a3fc19cfb7cba53533edbb98a4a,onmt/tests/test_beam_search.py,TestBeamSearch,test_advance_with_all_repeats_gets_blocked,#TestBeamSearch#,30

Before Change

                        [0] + [-float("inf")] * (beam_sz - 1))\
                        .repeat(batch_sz, 1)
                    expected_scores[:, :] = self.BLOCKED_SCORE
                    expected_scores = torch.tensor(
                        self.BLOCKED_SCORE).repeat(batch_sz, beam_sz)

    def test_advance_with_some_repeats_gets_blocked(self):
        // beam 0 and beam >=2 will repeat (beam >= 2 repeat dummy scores)
        beam_sz = 5

After Change

                    expected_scores = torch.tensor(
                        [self.BLOCKED_SCORE] + [-float("inf")] * (beam_sz - 1)
                    ).repeat(batch_sz, 1)
                    self.assertTrue(beam.topk_log_probs.equal(expected_scores))

    def test_advance_with_some_repeats_gets_blocked(self):
        // beam 0 and beam >=2 will repeat (beam >= 2 repeat dummy scores)
        beam_sz = 5

Instances