increased iters to 2, for more thorough testing

ReX-XAI · Dec 10, 2024 · 671df00 · 671df00
1 parent aec7510
commit 671df00
Show file tree

Hide file tree

Showing 4 changed files with 12 additions and 12 deletions.
diff --git a/tests/__snapshots__/_explanation_onnx_test.ambr b/tests/__snapshots__/_explanation_onnx_test.ambr
@@ -25,7 +25,7 @@
       heatmap_colours='magma',
       info=True,
       insertion_step=100,
-      iters=1,
+      iters=2,
       mark_segments=False,
       mask_value=0,
       means=None,
@@ -106,13 +106,13 @@
            [False, False, False,  ..., False, False, False],
            [False, False, False,  ..., False, False, False],
            [False, False, False,  ..., False, False, False]]]),
-    maps={209: 3},
+    maps={209: 4},
     mask_func=0,
     run_stats=dict({
-      'avg_box_size': 1731.3333333333333,
+      'avg_box_size': 2747.4166666666665,
       'max_depth_reached': 4,
-      'total_failing': 88,
-      'total_passing': 24,
+      'total_failing': 169,
+      'total_passing': 41,
     }),
     target=CLASS: 209, CONF: 0.391777366399765, TARGET_CLASS: None, TARGET_CONFIDENCE: None,
   )

diff --git a/tests/__snapshots__/_explanation_test.ambr b/tests/__snapshots__/_explanation_test.ambr
@@ -23,7 +23,7 @@
       heatmap_colours='magma',
       info=True,
       insertion_step=100,
-      iters=1,
+      iters=2,
       mark_segments=False,
       mask_value=0,
       means=None,
@@ -105,13 +105,13 @@
            [False, False, False,  ..., False, False, False],
            [False, False, False,  ..., False, False, False],
            [False, False, False,  ..., False, False, False]]]),
-    maps={207: 3},
+    maps={207: 4},
     mask_func=0,
     run_stats=dict({
-      'avg_box_size': 216.0,
+      'avg_box_size': 183.5625,
       'max_depth_reached': 5,
-      'total_failing': 71,
-      'total_passing': 41,
+      'total_failing': 126,
+      'total_passing': 84,
     }),
     target=CLASS: 207, CONF: 0.2046184241771698, TARGET_CLASS: None, TARGET_CONFIDENCE: None,
   )

diff --git a/tests/_explanation_onnx_test.py b/tests/_explanation_onnx_test.py
@@ -11,7 +11,7 @@
 def args_onnx(resnet50):
     args = CausalArgs()
     args.path = "imgs/dog.jpg"
-    args.iters = 1
+    args.iters = 2
     args.search_limit = 1000
     args.seed = 100
     args.model = resnet50

diff --git a/tests/_explanation_test.py b/tests/_explanation_test.py
@@ -12,7 +12,7 @@
 def args_custom_script():
     args = CausalArgs()
     args.path = "imgs/dog.jpg"
-    args.iters = 1
+    args.iters = 2
     args.search_limit = 1000
     args.seed = 42
     process_custom_script("tests/scripts/pytorch.py", args)