Method test_lm_decode

lm_inference_test.py:65–74

(self)

    )

  def test_lm_decode(self):
    outputs = LmInferenceTest.loaded_lm.beam_decode(
        '{S} a : ; b : ; c : ; d : T a b c d 00 T a c b d 01 ? T a d b c'
        ' {F1} x00',
        eos_tokens=[';'],
    )
    self.assertEqual(
        outputs['seqs_str'],
        ['e : D a b c e 02 D a c b e 03 ;', 'e : C a c e 02 C b d e 03 ;'],
    )

  def test_lm_score_may_fail_numerically_for_external_meliad(self):
    outputs = LmInferenceTest.loaded_lm.beam_decode(