From bee416ef29ffa9e68f222485fef27cad9f9c7adc Mon Sep 17 00:00:00 2001 From: Benedikt Johannes Date: Sun, 22 Feb 2026 14:38:46 +0100 Subject: [PATCH] Follow-up for Fixing SSIM Test Method Names Signed-off-by: Benedikt Johannes --- tests/metrics/test_ssim_metric.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/metrics/test_ssim_metric.py b/tests/metrics/test_ssim_metric.py index d79107e999..f1b62f916f 100644 --- a/tests/metrics/test_ssim_metric.py +++ b/tests/metrics/test_ssim_metric.py @@ -21,7 +21,7 @@ class TestSSIMMetric(unittest.TestCase): - def test2d_gaussian(self): + def test_2d_gaussian(self): set_determinism(0) preds = torch.abs(torch.randn(2, 3, 16, 16)) target = torch.abs(torch.randn(2, 3, 16, 16)) @@ -34,7 +34,7 @@ def test2d_gaussian(self): expected_value = 0.045415 self.assertTrue(expected_value - result.item() < 0.000001) - def test2d_uniform(self): + def test_2d_uniform(self): set_determinism(0) preds = torch.abs(torch.randn(2, 3, 16, 16)) target = torch.abs(torch.randn(2, 3, 16, 16)) @@ -47,7 +47,7 @@ def test2d_uniform(self): expected_value = 0.050103 self.assertTrue(expected_value - result.item() < 0.000001) - def test3d_gaussian(self): + def test_3d_gaussian(self): set_determinism(0) preds = torch.abs(torch.randn(2, 3, 16, 16, 16)) target = torch.abs(torch.randn(2, 3, 16, 16, 16))