Skip to content

Commit cc6e3cf

Browse files
committed
Retain old data type for Contour init args
1 parent 3510428 commit cc6e3cf

File tree

1 file changed

+7
-1
lines changed

1 file changed

+7
-1
lines changed

src/python/model_api/models/result/segmentation.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -149,7 +149,13 @@ def rotated_rects(self, value):
149149

150150

151151
class Contour:
152-
def __init__(self, label: str, probability: float, shape: np.ndarray, child_shapes: list[np.ndarray] | None = None):
152+
def __init__(
153+
self,
154+
label: str,
155+
probability: float,
156+
shape: np.ndarray | list[tuple[int, int]],
157+
child_shapes: list[np.ndarray] | list[tuple[int, int]] | None = None,
158+
):
153159
self.shape = shape
154160
self.label = label
155161
self.probability = probability

0 commit comments

Comments
 (0)