Fix wrong error message. (#1379)

Closes #1342 Co-authored-by: Vladimir Iglovikov <ternaus@users.noreply.github.com>
albumentations-team · Feb 12, 2023 · 6543733 · 6543733
1 parent 1eceb79
commit 6543733
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/albumentations/core/composition.py b/albumentations/core/composition.py
@@ -285,7 +285,7 @@ def _check_args(self, **kwargs) -> None:
         if self.is_check_shapes and shapes and shapes.count(shapes[0]) != len(shapes):
             raise ValueError(
                 "Height and Width of image, mask or masks should be equal. You can disable shapes check "
-                "by calling disable_shapes_check method of Compose class (do it only if you are sure "
+                "by setting a parameter is_check_shapes=False of Compose class (do it only if you are sure "
                 "about your data consistency)."
             )
 

diff --git a/tests/test_core.py b/tests/test_core.py
@@ -429,7 +429,7 @@ def test_compose_image_mask_equal_size(targets):
 
     assert str(exc_info.value).startswith(
         "Height and Width of image, mask or masks should be equal. "
-        "You can disable shapes check by calling disable_shapes_check method "
+        "You can disable shapes check by setting a parameter is_check_shapes=False "
         "of Compose class (do it only if you are sure about your data consistency)."
     )
     # test after disabling shapes check