fixing tests

cvat-ai · Feb 13, 2025 · 35c9d3c · 35c9d3c
1 parent c5de4ff
commit 35c9d3c
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/cvat/apps/lambda_manager/views.py b/cvat/apps/lambda_manager/views.py
@@ -836,7 +836,9 @@ def parse_anno(anno, labels):
                 elif anno["type"] == "mask":
                     [xtl, ytl, xbr, ybr] = shape["points"][-4:]
                     cut_points = shape["points"][:-4]
-                    rle = mask_tools.mask_to_rle(np.array(cut_points))["counts"].tolist()
+                    rle = mask_tools.mask_to_rle(np.array(cut_points)[:, np.newaxis])[
+                        "counts"
+                    ].tolist()
                     rle.extend([xtl, ytl, xbr, ybr])
                     shape["points"] = rle