diff --git a/dm_pix/_src/metrics_test.py b/dm_pix/_src/metrics_test.py index c3e6b09..4b1b763 100644 --- a/dm_pix/_src/metrics_test.py +++ b/dm_pix/_src/metrics_test.py @@ -140,7 +140,7 @@ def test_ssim_golden(self): np.testing.assert_allclose(ssim, ssim_gt, atol=1e-5, rtol=1e-5) else: np.testing.assert_allclose( - np.mean(ssim, list(range(-3, 0))), ssim_gt, atol=1e-5, rtol=1e-5 + np.mean(ssim, list(range(-3, 0))), ssim_gt, atol=1e-5, rtol=1e-5 # pyrefly: ignore[no-matching-overload] ) self.assertLessEqual(np.max(ssim), 1.0) self.assertGreaterEqual(np.min(ssim), -1.0)