Skip to content

Commit 57a2d75

Browse files
authored
Removed shape check from mask
1 parent b8a0e76 commit 57a2d75

File tree

1 file changed

+0
-8
lines changed

1 file changed

+0
-8
lines changed

hub/schema/mask.py

Lines changed: 0 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,6 @@ def __init__(
4242
If default value is chosen, automatically detects how to split into chunks
4343
4444
"""
45-
self._check_shape(shape)
4645
super().__init__(
4746
shape,
4847
"bool_",
@@ -51,13 +50,6 @@ def __init__(
5150
compressor=compressor,
5251
)
5352

54-
def _check_shape(self, shape):
55-
"""Check if provided shape maches mask characteristics."""
56-
if len(shape) != 3 or shape[-1] != 1:
57-
raise ValueError(
58-
"Wrong mask shape provided, should be of the format (height, width, 1) where height and width are integers or None"
59-
)
60-
6153
def __str__(self):
6254
out = super().__str__()
6355
out = "Mask" + out[6:]

0 commit comments

Comments
 (0)