From 7b85948e416abe3b6aea6bf6f97ae130be477834 Mon Sep 17 00:00:00 2001 From: Joseph Paul Cohen Date: Thu, 2 Jan 2025 17:39:34 -0800 Subject: [PATCH] change to interpolate and refactor --- tests/test_models.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/test_models.py b/tests/test_models.py index fbb9129..b43e00f 100644 --- a/tests/test_models.py +++ b/tests/test_models.py @@ -92,12 +92,12 @@ def test_normalization_check(): test_x[0][0][1] = ra[1] xrv.models.warning_log = {} model(test_x) - assert xrv.models.warning_log['norm_correct'] == False, ra + assert xrv.utils.warning_log['norm_correct'] == False, ra for ra in correct_ranges: test_x = torch.zeros([1,1,224,224]) test_x.uniform_(ra[0], ra[1]) xrv.models.warning_log = {} model(test_x) - assert xrv.models.warning_log['norm_correct'] == True, ra + assert xrv.utils.warning_log['norm_correct'] == True, ra