From d453ea61205882eb4210f8c4d2fb001e2c96872b Mon Sep 17 00:00:00 2001 From: Younes Belkada <49240599+younesbelkada@users.noreply.github.com> Date: Tue, 14 Jun 2022 18:14:12 +0200 Subject: [PATCH] fix tolerance for a bloom slow test (#17634) --- tests/models/bloom/test_modeling_bloom.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/models/bloom/test_modeling_bloom.py b/tests/models/bloom/test_modeling_bloom.py index 14b711c44f..f71618eae8 100644 --- a/tests/models/bloom/test_modeling_bloom.py +++ b/tests/models/bloom/test_modeling_bloom.py @@ -723,7 +723,7 @@ class BloomEmbeddingTest(unittest.TestCase): } if cuda_available: - self.assertEqual(MEAN_VALUE_LAST_LM, logits.last_hidden_state.mean().item()) + self.assertAlmostEqual(MEAN_VALUE_LAST_LM, logits.last_hidden_state.mean().item(), places=4) else: self.assertAlmostEqual(MEAN_VALUE_LAST_LM, logits.last_hidden_state.mean().item(), places=3)