reverting test_dict values for other test with hardcoded values

microsoft · Jan 18, 2022 · f33cd65 · f33cd65
1 parent e25b41e
commit f33cd65
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/Tests/ML/histopathology/utils/test_metrics_utils.py b/Tests/ML/histopathology/utils/test_metrics_utils.py
@@ -44,7 +44,7 @@ def assert_equal_lists(pred: List, expected: List) -> None:
              ResultsKey.PROB: [Tensor([0.5]), Tensor([0.7]), Tensor([0.4]), Tensor([1.0])],
              ResultsKey.TRUE_LABEL: [0, 1, 1, 1],
              ResultsKey.BAG_ATTN:
-                  [Tensor([[0.1, 0.3, 0.5, 0.8]]),
+                  [Tensor([[0.1, 0.0, 0.2, 0.15]]),
                   Tensor([[0.10, 0.18, 0.15, 0.13]]),
                   Tensor([[0.25, 0.23, 0.20, 0.21]]),
                   Tensor([[0.33, 0.31, 0.37, 0.35]])],

diff --git a/Tests/ML/test_data/histo_heatmaps/heatmap_overlay.png b/Tests/ML/test_data/histo_heatmaps/heatmap_overlay.png