We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent bee416e commit 86621f9Copy full SHA for 86621f9
1 file changed
tests/metrics/test_ssim_metric.py
@@ -32,7 +32,7 @@ def test_2d_gaussian(self):
32
metric(preds, target)
33
result = metric.aggregate()
34
expected_value = 0.045415
35
- self.assertTrue(expected_value - result.item() < 0.000001)
+ self.assertTrue(abs(expected_value - result.item()) < 0.000001)
36
37
def test_2d_uniform(self):
38
set_determinism(0)
0 commit comments