Skip to content

Commit 2b2ab97

Browse files
committed
Fix returning deleted images
1 parent ad3fed7 commit 2b2ab97

File tree

2 files changed

+12
-7
lines changed

2 files changed

+12
-7
lines changed

superannotate/db/images.py

Lines changed: 11 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -159,13 +159,18 @@ def get_image_metadata(project, image_names):
159159
json_req=json_req,
160160
)
161161

162-
metadata = response.json()
163-
if len(metadata) == 0:
162+
metadata_raw = response.json()
163+
metadata_without_deleted = []
164+
for im_metadata in metadata_raw:
165+
if 'deleted' in im_metadata and im_metadata['deleted'] == 1:
166+
continue
167+
metadata_without_deleted.append(im_metadata)
168+
if len(metadata_without_deleted) == 0:
164169
raise SABaseException(
165170
0,
166171
f"None of the images in {image_names} exist in the provided project"
167172
)
168-
for item in metadata:
173+
for item in metadata_without_deleted:
169174
item['annotation_status'] = common.annotation_status_int_to_str(
170175
item['annotation_status']
171176
)
@@ -178,9 +183,9 @@ def get_image_metadata(project, image_names):
178183
item['segmentation_status']
179184
)
180185

181-
if len(metadata) == 1:
182-
return metadata[0]
183-
return metadata
186+
if len(metadata_without_deleted) == 1:
187+
return metadata_without_deleted[0]
188+
return metadata_without_deleted
184189

185190

186191
def set_image_annotation_status(project, image_name, annotation_status):

superannotate/version.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
__version__ = "3.2.4"
1+
__version__ = "3.2.5"

0 commit comments

Comments
 (0)