From ac918f10298e74733446fd46c19395b88e9be5a6 Mon Sep 17 00:00:00 2001 From: Jannis Vamvas Date: Wed, 27 Dec 2023 11:39:14 +0100 Subject: [PATCH] Remove test condition that depends on randomness --- tests/test_generate.py | 1 - 1 file changed, 1 deletion(-) diff --git a/tests/test_generate.py b/tests/test_generate.py index e758373..ad5a31f 100644 --- a/tests/test_generate.py +++ b/tests/test_generate.py @@ -102,7 +102,6 @@ def test_model_output_extended(self): torch.testing.assert_close(output.sequences[0], sample.sequences[0][:output.sequences[0].shape[0]]) else: torch.testing.assert_close(output.sequences[0][:sample.sequences[0].shape[0]], sample.sequences[0]) - torch.testing.assert_close(output.sequences[0], sample.sequences[0]) self.assertIsNotNone(sample.scores) self.assertEqual(1, len(sample.scores[0])) self.assertIsNotNone(sample.attentions)