test(losses): change assertEqual to assertAlmostEqual

Failed in CI with: AssertionError: 1.401298464324817e-45 != 0.0
2024-05-01 14:28:55 +02:00 · 2024-05-01 14:28:55 +02:00 · 98e21d0f02
parent 962f9bbbcf
commit 98e21d0f02
1 changed files with 1 additions and 1 deletions
--- 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)