From b1ba03e082e5a3558100f453c49be4b26e8d2a63 Mon Sep 17 00:00:00 2001 From: Sylvain Gugger Date: Wed, 9 Feb 2022 12:06:59 -0500 Subject: [PATCH] Fix quality --- tests/test_generation_beam_search.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_generation_beam_search.py b/tests/test_generation_beam_search.py index 339b6f62be..ec6a94e0b3 100644 --- a/tests/test_generation_beam_search.py +++ b/tests/test_generation_beam_search.py @@ -318,7 +318,7 @@ class ConstrainedBeamSearchTester: beam_hyp.add(input_ids[beam_idx], -10.0 + float(beam_idx)) # -10.0 is removed => -9.0 is worst score - self.parent.assertAlmostEqual(beam_hyp.worst_score, -9.0 / (self.sequence_length ** beam_hyp.length_penalty)) + self.parent.assertAlmostEqual(beam_hyp.worst_score, -9.0 / (self.sequence_length**beam_hyp.length_penalty)) # -5.0 is better than worst score => should not be finished self.parent.assertFalse(beam_hyp.is_done(-5.0, self.sequence_length))