refactoring

prime-slam · Mar 4, 2024 · c737393 · c737393
1 parent 9854817
commit c737393
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/tests/utils/test_sam_mask.py b/tests/utils/test_sam_mask.py
@@ -157,8 +157,7 @@ def test_find_union_mask(
 ):
     actual_union_mask = find_union_mask(mask1, mask2)
     assert (
-        actual_union_mask["segmentation"]
-        == expected_union_mask["segmentation"]
+        actual_union_mask["segmentation"] == expected_union_mask["segmentation"]
     ).all()
     assert actual_union_mask["bbox"] == expected_union_mask["bbox"]
     assert actual_union_mask["area"] == expected_union_mask["area"]