fix test

mlmed · Jan 3, 2025 · 7a8c3d9 · 7a8c3d9
1 parent 7b85948
commit 7a8c3d9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_models.py b/tests/test_models.py
@@ -90,14 +90,14 @@ def test_normalization_check():
             # so here the first 2 pixels are set to the limits
             test_x[0][0][0] = ra[0]  
             test_x[0][0][1] = ra[1]
-            xrv.models.warning_log = {}
+            xrv.utils.warning_log = {}
             model(test_x)
             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 = {}
+            xrv.utils.warning_log = {}
             model(test_x)
             assert xrv.utils.warning_log['norm_correct'] == True, ra