Remove test condition that depends on randomness

ZurichNLP · Dec 27, 2023 · ac918f1 · ac918f1
1 parent 935089e
commit ac918f1
Showing 1 changed file with 0 additions and 1 deletion.
diff --git 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)