From 98e21d0f025bb49ca345db706ebea7af8ea654c2 Mon Sep 17 00:00:00 2001 From: Enno Hermann Date: Wed, 1 May 2024 14:28:55 +0200 Subject: [PATCH] test(losses): change assertEqual to assertAlmostEqual Failed in CI with: AssertionError: 1.401298464324817e-45 != 0.0 --- tests/tts_tests/test_losses.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/tts_tests/test_losses.py b/tests/tts_tests/test_losses.py index 522b7bb1..794478dc 100644 --- a/tests/tts_tests/test_losses.py +++ b/tests/tts_tests/test_losses.py @@ -216,7 +216,7 @@ class BCELossTest(unittest.TestCase): late_x = -200.0 * sequence_mask(length + 1, 100).float() + 100.0 # simulate logits on late stopping loss = layer(true_x, target, length) - self.assertEqual(loss.item(), 0.0) + self.assertAlmostEqual(loss.item(), 0.0) loss = layer(early_x, target, length) self.assertAlmostEqual(loss.item(), 2.1053, places=4)