Skip to content

Commit 4ea760c

Browse files
authored
Merge pull request #709 from superannotateai/fix_numpy
fix numpay alltrue>all, fix tests
2 parents 377ba56 + 87ff4be commit 4ea760c

File tree

3 files changed

+1
-45
lines changed

3 files changed

+1
-45
lines changed

src/superannotate/lib/core/usecases/images.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -544,7 +544,7 @@ def execute(self):
544544
fill_color = *class_color_map[annotation["className"]], 255
545545
for part in annotation["parts"]:
546546
part_color = *self.generate_color(part["color"]), 255
547-
temp_mask = np.alltrue(annotation_mask == part_color, axis=2)
547+
temp_mask = np.all(annotation_mask == part_color, axis=2)
548548
empty_image_arr[temp_mask] = fill_color
549549

550550
images = [

tests/integration/annotations/test_annotation_delete.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -47,7 +47,6 @@ def test_delete_annotations(self):
4747
"name": "example_image_1.jpg",
4848
"height": None,
4949
"width": None,
50-
"isPredicted": False,
5150
"status": "NotStarted",
5251
"pinned": False,
5352
"annotatorEmail": None,
@@ -77,7 +76,6 @@ def test_delete_annotations_by_name(self):
7776
"name": "example_image_1.jpg",
7877
"height": None,
7978
"width": None,
80-
"isPredicted": False,
8179
"status": "NotStarted",
8280
"pinned": False,
8381
"annotatorEmail": None,

tests/integration/test_ml_funcs.py

Lines changed: 0 additions & 42 deletions
This file was deleted.

0 commit comments

Comments
 (0)