From af8a1bebc6b86564bba55551a959ee48f6c767a0 Mon Sep 17 00:00:00 2001 From: viknesh-ai Date: Sat, 22 Aug 2026 15:55:18 +0530 Subject: [PATCH] Migrate tests/test_gpu_aamp_stimp.py from npt.assert_almost_equal to npt.assert_allclose --- tests/test_gpu_aamp_stimp.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/test_gpu_aamp_stimp.py b/tests/test_gpu_aamp_stimp.py index 88f5a656d..1950d901f 100644 --- a/tests/test_gpu_aamp_stimp.py +++ b/tests/test_gpu_aamp_stimp.py @@ -59,7 +59,7 @@ def test_gpu_aamp_stimp(T): naive.replace_inf(ref_PAN) naive.replace_inf(cmp_PAN) - npt.assert_almost_equal(ref_PAN, cmp_PAN) + npt.assert_allclose(cmp_PAN, ref_PAN, atol=1.5e-07) # Compare transformed pan cmp_pan = pan.PAN_ @@ -76,4 +76,4 @@ def test_gpu_aamp_stimp(T): naive.replace_inf(ref_pan) naive.replace_inf(cmp_pan) - npt.assert_almost_equal(ref_pan, cmp_pan) + npt.assert_allclose(cmp_pan, ref_pan, atol=1.5e-07)